Update app.py

app.py

@@ -7,7 +7,6 @@ import requests.exceptions
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.repocard import metadata_load
 
-
 app = gr.Blocks()
 
 model_id_1 = "nlptown/bert-base-multilingual-uncased-sentiment"
@@ -16,7 +15,6 @@ model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
 model_id_4 = "lordtt13/emo-mobilebert"
 model_id_5 = "juliensimon/reviews-sentiment-analysis"
 
-
 def get_prediction(model_id):
 
     classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
@@ -27,7 +25,6 @@ def get_prediction(model_id):
             return prediction
     return predict
 
-
 with app:
     gr.Markdown(
     """
@@ -44,15 +41,43 @@ with app:
     """)
     with gr.Row():
         with gr.Column():
-            text1 = gr.Textbox(label="Model 1 = nlptown/bert-base-multilingual-uncased-sentiment")
+            gr.Markdown(
+            """
+            Model 1 = nlptown/bert-base-multilingual-uncased-sentiment
+            """)
             btn1 = gr.Button("Predict - Model 1")
-            text2 = gr.Textbox(label="Model 2 = microsoft/deberta-base")
+            gr.Markdown(
+            """
+            Model 2 = microsoft/deberta-base
+            """)
             btn2 = gr.Button("Predict - Model 2")
+            gr.Markdown(
+            """
+            Model 3 = distilbert-base-uncased-finetuned-sst-2-english"
+            """)
+            btn3 = gr.Button("Predict - Model 3")
+            gr.Markdown(
+            """
+            Model 4 = lordtt13/emo-mobilebert
+            """)
+            btn4 = gr.Button("Predict - Model 4")
+            gr.Markdown(
+            """
+            Model 5 = juliensimon/reviews-sentiment-analysis
+            """)
+            btn5 = gr.Button("Predict - Model 5")
+
         with gr.Column():
             out_1 = gr.Textbox(label="Predictions for Model 1")
-            out_2 = gr.Textbox(label="Predictions for Model 2")     
+            out_2 = gr.Textbox(label="Predictions for Model 2")  
+            out_3 = gr.Textbox(label="Predictions for Model 3")        
+            out_4 = gr.Textbox(label="Predictions for Model 4")     
+            out_5 = gr.Textbox(label="Predictions for Model 5")     
+
         btn1.click(fn=get_prediction(model_id_1), inputs=inp_1, outputs=out_1)
         btn2.click(fn=get_prediction(model_id_2), inputs=inp_1, outputs=out_2)
-    
+        btn3.click(fn=get_prediction(model_id_3), inputs=inp_1, outputs=out_3)
+        btn4.click(fn=get_prediction(model_id_4), inputs=inp_1, outputs=out_4)
+        btn5.click(fn=get_prediction(model_id_5), inputs=inp_1, outputs=out_5)
    
 app.launch()