Update app.py

app.py

@@ -7,12 +7,20 @@ model2 = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
 
 def predict_sentiment(text, model_choice):
     try:
-        if model_choice == "Model 1 (RoBERTa-large)":
+        if model_choice == "Model 1 (Default Transformer)":
             predictions = model1(text)
-        elif model_choice == "Model 2 (BERTweet-base)":
+        elif model_choice == "Model 2 (BERTweet)":
             predictions = model2(text)
-        return f"Label: {predictions[0]['label']}, Score: {predictions[0]['score']:.4f}"
+        else:
+            return "Model choice is not recognized."
+
+        if predictions:
+            return f"Label: {predictions[0]['label']}, Score: {predictions[0]['score']:.4f}"
+        else:
+            return "No predictions returned by the model."
+
     except Exception as e:
+        # This catches any error and returns it directly
         return f"Error processing input: {e}"
 
 def documentation():