Initial commit

app.py

@@ -26,7 +26,7 @@ def trait_classifier(text: str) -> Dict[str, float]:
                                         top_k=None) # return all possible scores (not just top-1)
     neu_classifier = pipeline(task="text-classification", 
                                         # Because our model is on Hugging Face already, we can pass in the model name directly
-                                        model="kar0lina/petarda_xlm-roberta-base_pandora1000-neu", # link to model on HF Hub
+                                        model="kar0lina/petarda-pandora1000-neu-xlm-roberta-base", # link to model on HF Hub
                                         device="cuda" if torch.cuda.is_available() else "cpu",
                                         top_k=None) # return all possible scores (not just top-1)
     ope_classifier = pipeline(task="text-classification",