Update app.py

app.py

@@ -11,6 +11,12 @@ import torch
 model_folder = "zjunlp/chatcell-small"
 tokenizer = AutoTokenizer.from_pretrained(model_folder)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_folder)
+if torch.cuda.is_available():
+    device = torch.device("cuda")
+    model.to(device)  
+else:
+    device = torch.device("cpu")
+
 model.eval()