Update app.py

app.py

@@ -11,7 +11,7 @@ best_clf = joblib.load("./saved/classifier_llama32.joblib")
 
 encoder = LLMEncoder.from_pretrained(
     "SengTak/lec_harmfulprompclassifierencoder",
-    device_map="cpu",
+    device_map="cuda" if torch.cuda.is_available() else "cpu",
     torch_dtype=torch.bfloat16,
     #torch_dtype=torch,
     #cache_dir=cache_dir