Update app.py

app.py

@@ -13,7 +13,7 @@ model = AutoModelForSequenceClassification.from_pretrained(
     model_path,
     num_labels=1,
     load_in_8bit=True,
-    device_map={"": torch.device(device)}
+    device_map={'': device(type='cuda', index=0)}
 )
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model.eval()