Update app.py

app.py

@@ -29,6 +29,29 @@ def get_ext(uploaded_file):
     extype = uID+uploaded_file.name
     return extype
 
+
+
+@st.cache_resource(ttl=3600)
+# def read_model(local_path):
+#     with open(local_path, "rb") as f:
+#         svm_classifier = pickle.load(f)
+#     return svm_classifier
+
+
+# def read_tf(local_path):
+#     with open(local_path, "rb") as f:
+#         preprocessing = pickle.load(f)
+#     return preprocessing
+
+
+def read_model(url):
+    response = requests.get(url)
+    open("temp.pkl", "wb").write(response.content)
+    with open("temp.pkl", "rb") as f:
+        svm_classifier = pickle.load(f)
+    return svm_classifier
+
+
 def read_tf(url):
     response = requests.get(url)
     open("temp.pkl", "wb").write(response.content)
@@ -61,38 +84,6 @@ def conv_txt(extype):
     return papers
 
 
-#===Read data===
-
-# uploaded_file = st.file_uploader("Choose a file", type=['csv'])
-
-# df = None  # Initialize df as None outside the if block
-
-# if uploaded_file is not None:
-#     df = pd.read_csv(uploaded_file, encoding='latin-1')
-#     st.dataframe(df)
-    
-#     # Tag the "Abstract" column with the corresponding categories
-#     df['category'] = df['Abstract'].apply(predict_category)
-#     # df['category'] = df['category'].str[2:-2]
-    
-#     st.dataframe(df)
-
-# st.sidebar.header("Download Results")
-# st.sidebar.text("Download the tagged results as a CSV file.")
-
-# if df is not None:
-#     csv = df.to_csv(index=False)
-#     b64 = base64.b64encode(csv.encode()).decode()
-#     st.sidebar.download_button(
-#         label="Download", 
-#         data=b64,
-#         file_name="results.csv",
-#         mime="text/csv"
-#     )
-# else:
-#     st.sidebar.text("Please upload a file first to enable the download button.")
-
-
 uploaded_file = st.file_uploader("Choose a file", type=['csv'])
 st.sidebar.header("Download Results")
 st.sidebar.text("Download the tagged results as a CSV file.")