Update app.py

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 app = Flask(__name__)
 
 import torch
+
 from torch import cuda
 from tqdm import tqdm
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast