Update app.py

app.py

@@ -12,7 +12,10 @@ app = gr.Blocks()
 
 model_id_1 = "nlptown/bert-base-multilingual-uncased-sentiment"
 model_id_2 = "microsoft/deberta-base"
-model_id_3 = "juliensimon/distilbert-amazon-shoe-reviews"
+model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
+model_id_4 = "lordtt13/emo-mobilebert"
+model_id_5 = "juliensimon/reviews-sentiment-analysis"
+
 
 def load_agent(model_id):
     """
@@ -72,9 +75,9 @@ with app:
     """)
 
     with gr.Row():
-        btn1 = gr.Button("Predict for Model 1")
+        btn1 = gr.Button("Predict - Model 1")
     with gr.Row():
-        out_1 = gr.Textbox(label="Prediction for Model 1")
+        out_1 = gr.Textbox(label="Predictions for Model 1")
     btn1.click(fn=get_prediction(model_id_1), inputs=inp_1, outputs=out_1)
 
     gr.Markdown(
@@ -83,20 +86,43 @@ with app:
     """)
 
     with gr.Row():
-        btn2 = gr.Button("Predict for Model 2")
+        btn2 = gr.Button("Predict - Model 2")
     with gr.Row():
-        out_2 = gr.Textbox(label="Prediction for Model 2")      
+        out_2 = gr.Textbox(label="Predictions for Model 2")      
     btn2.click(fn=get_prediction(model_id_2), inputs=inp_1, outputs=out_2)
 
     gr.Markdown(
     """
-    Model 3 = juliensimon/distilbert-amazon-shoe-reviews
+    Model 3 = distilbert-base-uncased-finetuned-sst-2-english"
     """)
 
     with gr.Row():
-        btn3 = gr.Button("Predict for Model 3")
+        btn3 = gr.Button("Predict - Model 3")
     with gr.Row():
-        out_3 = gr.Textbox(label="Prediction for Model 3")      
+        out_3 = gr.Textbox(label="Predictions for Model 3")      
     btn3.click(fn=get_prediction(model_id_3), inputs=inp_1, outputs=out_3)
+
+    gr.Markdown(
+    """
+    Model 4 = lordtt13/emo-mobilebert
+    """)
+
+    with gr.Row():
+        btn4 = gr.Button("Predict - Model 4")
+    with gr.Row():
+        out_4 = gr.Textbox(label="Predictions for Model 4")      
+    btn3.click(fn=get_prediction(model_id_4), inputs=inp_1, outputs=out_4)
+
+    gr.Markdown(
+    """
+    Model 5 = juliensimon/reviews-sentiment-analysis
+    """)
+
+    with gr.Row():
+        btn5 = gr.Button("Predict - Model 5")
+    with gr.Row():
+        out_5 = gr.Textbox(label="Predictions for Model 5")      
+    btn5.click(fn=get_prediction(model_id_4), inputs=inp_1, outputs=out_4)
+
    
 app.launch()