Update gradio_interface.py

gradio_interface.py

@@ -21,10 +21,12 @@ def save_to_file(text, predicted_labels):
 
 # Gradio Interface
 inputs = [
-    gr.Textbox(lines=7, label="Enter your text"),
-    gr.File(label="Upload a file")
+    gr.inputs.Textbox(lines=7, label="Enter your text"),
+    gr.inputs.File(label="Upload a file")
 ]
 
+output = gr.outputs.Dataframe()
+
 def classify_and_save(input_text, input_file):
     if input_text:
         text = input_text
@@ -37,5 +39,5 @@ def classify_and_save(input_text, input_file):
     save_to_file(text, predicted_labels)
     return predicted_labels
 
-iface = gr.Interface(fn=classify_and_save, inputs=inputs, title="Text Classifier")
+iface = gr.Interface(fn=classify_and_save, inputs=inputs, outputs=output, title="Text Classifier")
 iface.launch(share=True)