Update app.py

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 from io import StringIO
 
 def predict_and_download(classifier_oneclass_or_Cosine_distance ,positive_csv_file, unlabelled_csv_file, n,Hyperparameter_nu):
-    if choice == "oneclass" :
+    if classifier_oneclass_or_Cosine_distance == "oneclass" :
        selected_paper_info = oneclass.select_top_n_papers(n, positive_csv_file, unlabelled_csv_file,Hyperparameter_nu)
     else :    
        selected_paper_info = Cosine_distance.recommend_papers(positive_csv_file, positive_csv_file, n)
@@ -26,7 +26,7 @@ def predict_and_download(classifier_oneclass_or_Cosine_distance ,positive_csv_fi
 # Create the interface
 iface = gr.Interface(
     fn=predict_and_download,
-    inputs=["text","file", "file", "number","number"],
+    inputs=["text","file", "file", "number"],
     outputs=[gr.DataFrame(label="Recommended Papers"), gr.DownloadButton(label="Download CSV")],
     title="Personalized Arxiv Feed",
     description="Enter text and upload CSV files for labelled and unlabelled data.",