Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,12 +15,14 @@ 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 from your HF Model Repo
     # 1. Load DistilBERT Classifier from your HF Model Repo
     repo_id = "Shauryaaa05/AutoResolve-DistilBERT"
+    folder_name = "autoresolve_distilbert_final"
     
-    distil_tokenizer = DistilBertTokenizerFast.from_pretrained(repo_id)
-    distil_model = DistilBertForSequenceClassification.from_pretrained(repo_id)
-    
+    distil_tokenizer = DistilBertTokenizerFast.from_pretrained(repo_id, subfolder=folder_name)
+    distil_model = DistilBertForSequenceClassification.from_pretrained(repo_id, subfolder=folder_name)
+
     # 2. Knowledge Base & Retriever
     kb = [
         "Refund Policy: Customers are entitled to a full refund within 30 days of purchase. To process, verify the order number and issue the refund to the original payment method.",