Update app.py

app.py

@@ -9,6 +9,10 @@ import re
 from sklearn.utils.class_weight import compute_class_weight
 import numpy as np
 
+# ✅ Device setup
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(f"Using device: {device}")
+
 # Download label encoder from Hugging Face Hub
 label_encoder_path = hf_hub_download(repo_id="Fredaaaaaa/hybrid_model", filename="label_encoder.pkl")
 with open(label_encoder_path, 'rb') as f:
@@ -44,6 +48,8 @@ loss_fn = torch.nn.CrossEntropyLoss(weight=class_weights)
 # The rest of your code follows here...
 
 
+
+
 # Function to properly clean drug names
 def clean_drug_name(drug_name):
     if not drug_name: