Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import pipeline
 
 pipe = pipeline("text-classification", model="jonas/sdg_classifier_osdg")
+pipe2 = pipeline("text-classification", model="peter2000/xlm-roberta-base-finetuned-osdg")
 
 sdg_dict = {
 1:"SDG 1: End poverty in all its forms everywhere",
@@ -23,7 +24,8 @@ sdg_dict = {
 
 def predict(text):
     preds = pipe(text)[0]
-    return sdg_dict[int(preds["label"][0])], str(round(preds["score"], 5)*100) + " %"
+    preds2 = pipe(text)[0]
+    return sdg_dict[int(preds["label"][0])], str(round(preds["score"], 5)*100) + " %", sdg_dict[int(preds2["label"][0])], str(round(preds2["score"], 5)*100) + " %"
 
 gradio_ui = gr.Interface(
     fn=predict,
@@ -33,6 +35,8 @@ gradio_ui = gr.Interface(
         gr.inputs.Textbox(lines=5, label="Paste some text here"),
     ],
     outputs=[
+        gr.outputs.Textbox(label="SDG Label"),
+        gr.outputs.Textbox(label="Certainty"),
         gr.outputs.Textbox(label="SDG Label"),
         gr.outputs.Textbox(label="Certainty")
     ],