Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,27 +15,11 @@ st.markdown("This end-to-end LLM Agent classifies your IT issue, retrieves the r
 # --- 1. Load Models (Cached so they only load once) ---
 @st.cache_resource
 def load_pipeline():
-    # 1. Load DistilBERT Classifier
-    # Note: You must upload your 'autoresolve_distilbert_final' folder to the HF space!
-    import os
-    actual_path = ""
-    for root, dirs, files in os.walk('/app'):
-        if 'model.safetensors' in files:
-            actual_path = root
-            break
-    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)
-      
-   
+    # 1. Load DistilBERT Classifier from your HF Model Repo
+    repo_id = "Shauryaaa05/AutoResolve-DistilBERT"
+    
+    distil_tokenizer = DistilBertTokenizerFast.from_pretrained(repo_id)
+    distil_model = DistilBertForSequenceClassification.from_pretrained(repo_id)
     
     # 2. Knowledge Base & Retriever
     kb = [