Update app.py

app.py

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