Update src/streamlit_app.py

src/streamlit_app.py

@@ -17,8 +17,15 @@ st.markdown("This end-to-end LLM Agent classifies your IT issue, retrieves the r
 def load_pipeline():
     # 1. Load DistilBERT Classifier
     # Note: You must upload your 'autoresolve_distilbert_final' folder to the HF space!
-    distil_tokenizer = DistilBertTokenizerFast.from_pretrained("../autoresolve_distilbert_final")
-    distil_model = DistilBertForSequenceClassification.from_pretrained("../autoresolve_distilbert_final")
+    import os
+    actual_path = ""
+    for root, dirs, files in os.walk('/app'):
+        if 'model.safetensors' in files:
+            actual_path = root
+            break
+            
+    distil_tokenizer = DistilBertTokenizerFast.from_pretrained(actual_path)
+    distil_model = DistilBertForSequenceClassification.from_pretrained(actual_path)
     
     # 2. Knowledge Base & Retriever
     kb = [