Update app.py

app.py

@@ -93,7 +93,7 @@ max_length = st.sidebar.slider("Max Length", min_value=50, max_value=1024, value
 # Load the model and tokenizer
 @st.cache_resource
 def load_model_and_tokenizer():
-    model_path = "C:/Users/MC/Ollama_UI/models--gpt2"  # Path to the local model directory
+    model_path = "gpt2"  # Path to the local model directory
     
     # Load model and tokenizer
     tokenizer = AutoTokenizer.from_pretrained(model_path)  # Use GPT-2 tokenizer from Hugging Face
@@ -153,7 +153,7 @@ def save_chat_data(chat_data):
 
 # Function to handle uploaded text or PDF files and convert PDF to txt
 def handle_uploaded_file(uploaded_file):
-    dataset_dir = "C:/Users/MC/Ollama_UI/datasets"
+    dataset_dir = "./datasets"
     dataset_path = Path(dataset_dir) / f"{uploaded_file.name}.txt"
 
     # Check if the file is a PDF