Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,8 +2,6 @@
|
|
| 2 |
from os import O_ACCMODE
|
| 3 |
import gradio as gr
|
| 4 |
from transformers import pipeline
|
| 5 |
-
# from huggingface_hub import HfApi, hf_hub_download
|
| 6 |
-
# from huggingface_hub.repocard import metadata_load
|
| 7 |
|
| 8 |
app = gr.Blocks()
|
| 9 |
|
|
@@ -13,7 +11,7 @@ model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
|
|
| 13 |
model_id_4 = "lordtt13/emo-mobilebert"
|
| 14 |
model_id_5 = "juliensimon/reviews-sentiment-analysis"
|
| 15 |
|
| 16 |
-
def parse_output(output_json):
|
| 17 |
list_pred=[]
|
| 18 |
for i in range(len(output_json[0])):
|
| 19 |
label = output_json[0][i]['label']
|
|
@@ -75,11 +73,11 @@ with app:
|
|
| 75 |
btn5 = gr.Button("Predict - Model 5")
|
| 76 |
|
| 77 |
with gr.Column():
|
| 78 |
-
out_1 = gr.Textbox(label="Predictions for Model 1"
|
| 79 |
-
out_2 = gr.Textbox(label="Predictions for Model 2"
|
| 80 |
-
out_3 = gr.Textbox(label="Predictions for Model 3"
|
| 81 |
-
out_4 = gr.Textbox(label="Predictions for Model 4"
|
| 82 |
-
out_5 = gr.Textbox(label="Predictions for Model 5"
|
| 83 |
|
| 84 |
btn1.click(fn=get_prediction(model_id_1), inputs=inp_1, outputs=out_1)
|
| 85 |
btn2.click(fn=get_prediction(model_id_2), inputs=inp_1, outputs=out_2)
|
|
|
|
| 2 |
from os import O_ACCMODE
|
| 3 |
import gradio as gr
|
| 4 |
from transformers import pipeline
|
|
|
|
|
|
|
| 5 |
|
| 6 |
app = gr.Blocks()
|
| 7 |
|
|
|
|
| 11 |
model_id_4 = "lordtt13/emo-mobilebert"
|
| 12 |
model_id_5 = "juliensimon/reviews-sentiment-analysis"
|
| 13 |
|
| 14 |
+
def parse_output(output_json):
|
| 15 |
list_pred=[]
|
| 16 |
for i in range(len(output_json[0])):
|
| 17 |
label = output_json[0][i]['label']
|
|
|
|
| 73 |
btn5 = gr.Button("Predict - Model 5")
|
| 74 |
|
| 75 |
with gr.Column():
|
| 76 |
+
out_1 = gr.Textbox(label="Predictions for Model 1")
|
| 77 |
+
out_2 = gr.Textbox(label="Predictions for Model 2")
|
| 78 |
+
out_3 = gr.Textbox(label="Predictions for Model 3")
|
| 79 |
+
out_4 = gr.Textbox(label="Predictions for Model 4")
|
| 80 |
+
out_5 = gr.Textbox(label="Predictions for Model 5")
|
| 81 |
|
| 82 |
btn1.click(fn=get_prediction(model_id_1), inputs=inp_1, outputs=out_1)
|
| 83 |
btn2.click(fn=get_prediction(model_id_2), inputs=inp_1, outputs=out_2)
|