Update app.py

app.py

@@ -19,15 +19,15 @@ from transformers import pipeline
   
 
 
-sp_model ="JonatanGk/roberta-base-bne-finetuned-cyberbullying-spanish"
+sp_model = "JonatanGk/roberta-base-bne-finetuned-cyberbullying-spanish"
 ca_model = "JonatanGk/roberta-base-ca-finetuned-cyberbullying-catalan"
 sp_analysis = pipeline("text-classification", model=sp_model, tokenizer=sp_model)
 ca_analysis = pipeline("text-classification", model=ca_model, tokenizer=ca_model)
 
 def bullying_analysis(language, text):
-    if language == 'English':
+    if language == 'Spanish':
         results = sp_analysis(text)
-    elif language == 'French':
+    elif language == 'Catalan':
         results = ca_analysis(text)
     return results[0]["label"], round(results[0]["score"], 5)