Update app.py

app.py

@@ -1,5 +1,10 @@
 import streamlit as st
 import pandas as pd
+import CosineDistanceClustering
+import PU
+import bert_embeddings
+import oneclass
+
 # Function to recommend papers
 def recommend_papers(positive_df, unlabelled_df, model="tfidf"):
     pass
@@ -9,7 +14,7 @@ def main():
     st.title("ArXiv Feed Recommendations")
 
     # User input for model selection
-    model = st.selectbox("Select Model", ["TF-IDF", "Nearest Neighbors"])
+    model = st.selectbox("Select Model", ["CosineDistanceClustering", "PU","oneclass"])
 
     # Upload CSV files
     st.subheader("Upload Positive Labeled CSV")