Update app.py

app.py

@@ -5,11 +5,17 @@ import torch
 # Set page configuration
 st.set_page_config(page_title="Spam Detection", page_icon="📧")
 
-# Load the local fine-tuned model and tokenizer
-model_path = "./fine_tuned_model"
+# Use an absolute path to avoid ambiguity
+import os
+model_path = os.path.join(os.getcwd(), "fine_tuned_model")  # Ensure it's treated as a local directory
+
 model = DistilBertForSequenceClassification.from_pretrained(model_path)
 tokenizer = DistilBertTokenizerFast.from_pretrained(model_path)
 
+
+
+
+
 # Function to predict spam
 def predict_spam(text):
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)