Update app.py

app.py

@@ -10,8 +10,9 @@ model_id_2 = "microsoft/deberta-base"
 model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
 model_id_4 = "lordtt13/emo-mobilebert"
 model_id_5 = "juliensimon/reviews-sentiment-analysis"
+model_id_6 = "sbcBI/sentiment_analysis_model"
 
-def parse_output(output_json):  
+def parse_output(output_json):  # list of list of dicts
     list_pred=[]
     for i in range(len(output_json[0])):
         label = output_json[0][i]['label']
@@ -71,18 +72,25 @@ with app:
             Model 5 = juliensimon/reviews-sentiment-analysis
             """)
             btn5 = gr.Button("Predict - Model 5")
+            gr.Markdown(
+            """
+            Model 6 = sbcBI/sentiment_analysis_model
+            """)
+            btn6 = gr.Button("Predict - Model 6")
+
 
         with gr.Column():
             out_1 = gr.Textbox(label="Predictions for Model 1")
             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")     
+            out_5 = gr.Textbox(label="Predictions for Model 5")  
+            out_6 = gr.Textbox(label="Predictions for Model 6")
 
         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)
-   
+        btn6.click(fn=get_prediction(model_id_5), inputs=inp_1, outputs=out_6)
 app.launch()