Shauryaaa05 commited on
Commit
2d6ecee
·
verified ·
1 Parent(s): 4c0f86a

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -2
src/streamlit_app.py CHANGED
@@ -23,9 +23,11 @@ def load_pipeline():
23
  if 'model.safetensors' in files:
24
  actual_path = root
25
  break
26
-
 
27
  distil_tokenizer = DistilBertTokenizerFast.from_pretrained(actual_path)
28
- distil_model = DistilBertForSequenceClassification.from_pretrained(actual_path)
 
29
 
30
  # 2. Knowledge Base & Retriever
31
  kb = [
 
23
  if 'model.safetensors' in files:
24
  actual_path = root
25
  break
26
+ actual_path = "autoresolve_distilbert_final"
27
+
28
  distil_tokenizer = DistilBertTokenizerFast.from_pretrained(actual_path)
29
+ distil_model = DistilBertForSequenceClassification.from_pretrained(actual_path)
30
+
31
 
32
  # 2. Knowledge Base & Retriever
33
  kb = [