Update app.py

app.py

@@ -21,7 +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_map="auto",
+                                             device=device,
                                             trust_remote_code=True)
 model.eval()