Update app.py

app.py

@@ -28,6 +28,10 @@ def predict_class(text):
         st.error(f"Error during prediction: {e}")
         return None
 
+# Create a directory to store uploaded files
+uploaded_files_dir = "uploaded_files"
+os.makedirs(uploaded_files_dir, exist_ok=True)
+
 st.title("Paper Citation Classifier")
 
 option = st.radio("Select input type:", ("Text", "PDF"))
@@ -51,7 +55,7 @@ elif option == "PDF":
     uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
 
     if uploaded_file is not None:
-        file_path = os.path.join("uploaded_files", uploaded_file.name)
+        file_path = os.path.join(uploaded_files_dir, uploaded_file.name)
         with open(file_path, "wb") as f:
             f.write(uploaded_file.getbuffer())
         st.success("File uploaded successfully.")