Update app.py

app.py

@@ -18,11 +18,15 @@ MODEL_ID = "dejanseo/QDF"
 HF_TOKEN = os.getenv("HF_TOKEN")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=HF_TOKEN)
+# ← updated block here
+model = AutoModelForSequenceClassification.from_pretrained(
+    MODEL_ID,
+    token=HF_TOKEN,
+    low_cpu_mem_usage=False,
+    device_map=None
+)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
-
-# ← changed from model.to(device)
-model = model.to_empty(device)
+model.to(device)
 
 model.eval()