Update app.py

app.py

@@ -18,22 +18,10 @@ def classify_text(text):
         {"label": item["label"], "score": round(item["score"], 4)}
         for result in results for item in result
     ]
-    return formatted_results
+    output = {}
+    for i in range(len(formatted_results['labels'])):
+        output[formatted_results['labels'][i]] = prediction['scores'][i]
+    return output
 
-
-
-# Create the Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("# Text Classification with Hugging Face Transformers")
-    gr.Markdown("Enter text to classify using the model: **Karzan/user_profile_skills_model**.")
-    with gr.Row():
-        with gr.Column():
-            input_text = gr.Textbox(label="Input Text", lines=3, placeholder="Type something...")
-            classify_button = gr.Button("Classify")
-        with gr.Column():
-            output_text = gr.JSON(label="Classification Results")
-
-    classify_button.click(classify_text, inputs=input_text, outputs=output_text)
-
-# Launch the app
-demo.launch()
+demo = gr.Interface(fn=classify_text, inputs=[gr.Textbox(label="Input")], outputs=gr.Label(label="Classification"), title="Text Classification")
+demo.launch()