Update app.py

app.py

@@ -19,8 +19,10 @@ def classify_text(text):
         for result in results for item in result
     ]
     output = {}
-    for i in range(len(formatted_results[0])):
-        output[formatted_results[0][i]['label']] = formatted_results[0][i]['score']
+    formatted_results = formatted_results[0]
+    print(formatted_results)
+    for i in range(len(formatted_results)):
+        output[formatted_results[i]['label']] = formatted_results[i]['score']
     return output
 
 demo = gr.Interface(fn=classify_text, inputs=[gr.Textbox(label="Input")], outputs=gr.Label(label="Classification"), title="Text Classification")