Update app.py

app.py

@@ -21,8 +21,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained('habdine/Esm2Text-Base-v1-1', 
                                             trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained('habdine/Esm2Text-Base-v1-1',
-                                             device=device,
-                                            trust_remote_code=True)
+                                            trust_remote_code=True).to(device)
 model.eval()