updated prediction fn to return just one value

app.py

@@ -16,7 +16,7 @@ def upload_file(files):
 def predict_topic(input_text):
     prediction = pipeline_classification_topics(input_text)
     # predicted_label = prediction[0]['label']
-    return prediction
+    return prediction[0]
 
 # Build Gradio interface
 with gr.Blocks() as demo:
@@ -25,7 +25,7 @@ with gr.Blocks() as demo:
 
     # Input text to be reframed
     text = gr.Textbox(label="Text")
-    #predictions = gr.outputs.Textbox(label="Topic")
+    # predictions = gr.outputs.Textbox(label="Topic")
     predictions = gr.Label("Topic")
 
     # Trigger button for topic prediction