Shauryaaa05 commited on
Commit
5bfbd1d
·
verified ·
1 Parent(s): 4a33152

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -17,8 +17,8 @@ st.markdown("This end-to-end LLM Agent classifies your IT issue, retrieves the r
17
  def load_pipeline():
18
  # 1. Load DistilBERT Classifier
19
  # Note: You must upload your 'autoresolve_distilbert_final' folder to the HF space!
20
- distil_tokenizer = DistilBertTokenizerFast.from_pretrained("./autoresolve_distilbert_final")
21
- distil_model = DistilBertForSequenceClassification.from_pretrained("./autoresolve_distilbert_final")
22
 
23
  # 2. Knowledge Base & Retriever
24
  kb = [
 
17
  def load_pipeline():
18
  # 1. Load DistilBERT Classifier
19
  # Note: You must upload your 'autoresolve_distilbert_final' folder to the HF space!
20
+ distil_tokenizer = DistilBertTokenizerFast.from_pretrained("../autoresolve_distilbert_final")
21
+ distil_model = DistilBertForSequenceClassification.from_pretrained("../autoresolve_distilbert_final")
22
 
23
  # 2. Knowledge Base & Retriever
24
  kb = [