Update src/streamlit_app.py

src/streamlit_app.py

@@ -23,10 +23,18 @@ def load_pipeline():
         if 'model.safetensors' in files:
             actual_path = root
             break
-    actual_path = "autoresolve_distilbert_final"
+    import os
+
+# 1. Get the directory of the current script (the /src folder)
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+
+# 2. Go up one level to the root, then into the model folder
+    actual_path = os.path.join(current_dir, "..", "autoresolve_distilbert_final")
 
+# 3. Load the model using this strict absolute path
     distil_tokenizer = DistilBertTokenizerFast.from_pretrained(actual_path)
-    distil_model = DistilBertForSequenceClassification.from_pretrained(actual_path)       
+    distil_model = DistilBertForSequenceClassification.from_pretrained(actual_path)
+      
    
     
     # 2. Knowledge Base & Retriever