Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import pandas as pd
 from io import StringIO
 
-def predict_and_download(positive_csv_file, unlabelled_csv_file, n, text):
+def predict_and_download(positive_csv_file, unlabelled_csv_file, n):
     selected_paper_info = oneclass.select_top_n_papers(n, positive_csv_file, unlabelled_csv_file)
     
     # Create a StringIO object to store CSV data
@@ -16,15 +16,13 @@ def predict_and_download(positive_csv_file, unlabelled_csv_file, n, text):
     csv_content = csv_buffer.getvalue()
     
     
-    
-    
     # Return selected_paper_info and CSV content
-    return selected_paper_info, csv_content
+    return selected_paper_info, csv_buffer
 
 # Create the interface
 iface = gr.Interface(
     fn=predict_and_download,
-    inputs=["file", "file", "number", "textbox"],
+    inputs=["file", "file", "number"],
     outputs=[gr.DataFrame(label="Selected Papers"), gr.DownloadButton(label="Download CSV")],
     title="Paper Prediction",
     description="Enter text and upload CSV files for labelled and unlabelled data.",