Update app.py

app.py

@@ -7,7 +7,7 @@ import torch
 
 target_list = ["Playful", "Infuriating", "Sentimental", "Cynical", "Depressing", "Awe-inspiring", "Patriotic", "Begrudging", "Educational", "Hopeful",
 "Sarcastic", "Disrespectful", "Disparaging"]
-device = torch.device('cuda' if torch.cuda.is_available() else 'CPU')
+#device = torch.device('cuda' if torch.cuda.is_available() else 'CPU')
 
 model_name = "valurank/finetuned-distilbert-multi-label-emotion"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -20,7 +20,7 @@ def get_probs(logits, threshold=0.5):
 
 def multi_label_emotions(text):
   inputs = tokenizer(text, return_tensors="pt", truncation=True)
-  model.to(device)
+  #model.to(device)
     
   with torch.no_grad():
       logits = model(**inputs).logits