Update app.py

app.py

@@ -25,7 +25,7 @@ def process_csv(file):
         
         # Get predicted classes
         _, predicted_classes = torch.max(outputs.logits, dim=1)
-        df['predicted_class'] = predicted_classes.numpy()"]
+        df['predicted_class'] = predicted_classes.numpy()
         
         # Return the processed DataFrame as a CSV string
         return df.to_csv(index=False)
@@ -35,12 +35,12 @@ def process_csv(file):
 
 # Create the Gradio interface
 input_csv = gr.File(label="Upload CSV File")
-output_csv = gr.File(label="Download Processed CSV")
+output_csv = gr.Textbox(label="Output")
 
 demo = gr.Interface(
     fn=process_csv,
     inputs=input_csv,
-    outputs=gr.Textbox(label="Output"),
+    outputs=output_csv,
     title="CSV Data Processing with TinyBERT",
     description="Upload a CSV file with a 'text' column, and the model will process the data and provide predictions."
 )