Update app.py

app.py

@@ -21,12 +21,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = AutoModelForSequenceClassification.from_pretrained(
     MODEL_ID,
     token=HF_TOKEN,
-    low_cpu_mem_usage=True,
-    device_map="auto"  # delegates placement automatically
-)
+    low_cpu_mem_usage=True
+).eval()
+
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
-model.to(device)
 
 model.eval()