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)
@@ -40,7 +40,7 @@ output_csv = gr.File(label="Download Processed CSV")
 demo = gr.Interface(
     fn=process_csv,
     inputs=input_csv,
-    outputs=output_csv,
+    outputs=gr.Textbox(label="Output"),
     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."
 )