Update app.py

app.py

@@ -44,8 +44,8 @@ def run_pipeline(text):
     sparse_result = dict(sparse_output)
     sparse_end = time.perf_counter()
     sparse_duration = (sparse_end - sparse_start) * 1000.0
-    
-    return {"politics":0.20, "public":0.80}, sparse_duration
+    dict_r = {sparse_result['labels'][0]:sparse_result['scores'][0],sparse_result['labels'][1]:sparse_result['scores'][1], sparse_result['labels'][2]:sparse_result['scores'][2]}
+    return dict_r, sparse_duration
 
 
 with gr.Blocks() as demo:
@@ -61,7 +61,7 @@ with gr.Blocks() as demo:
             btn = gr.Button("Submit")
             
             sparse_answers = gr.Label(label="Sparse model answers",
-                                    num_top_classes=2
+                                    num_top_classes=3
                                      )
             sparse_duration = gr.Number(label="Sparse Latency (ms):")
             gr.Examples([["The senate passed 3 laws today"],["Who are you voting for in 2020?"],["Public health is very important"]],inputs=[text],)