Update app.py

app.py

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