Update app.py

app.py

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