Update app.py

app.py

@@ -7,9 +7,9 @@ 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_name = "valurank/finetuned-distilbert-multi-label-emotion"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -46,7 +46,7 @@ def multi_label_emotions(text):
 
 demo = gr.Interface(multi_label_emotions, inputs=gr.Textbox(),
                     outputs = gr.Label(num_top_classes=16),
-                    title='Multi-label-emotion-classification')
+                    title="Multi-label-emotion-classification")
 
-if __name__ == '__main__':
+if __name__ == "__main__":
   demo.launch(debug=True)