HEHEBOIOG commited on
Commit
87dafb9
1 Parent(s): f11ecb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -22,12 +22,8 @@ except UnicodeDecodeError:
22
  vectorizer = TfidfVectorizer(stop_words='english')
23
  X_tfidf = vectorizer.fit_transform(medical_df['Questions'])
24
 
25
-
26
- # Load TinyLlama-15M model and tokenizer
27
- model_name = "nickypro/tinyllama-15M"
28
- tokenizer = AutoTokenizer.from_pretrained(model_name)
29
- model = AutoModelForCausalLM.from_pretrained(model_name)
30
-
31
 
32
  # Load pre-trained Sentence Transformer model
33
  sbert_model_name = "paraphrase-MiniLM-L6-v2"
 
22
  vectorizer = TfidfVectorizer(stop_words='english')
23
  X_tfidf = vectorizer.fit_transform(medical_df['Questions'])
24
 
25
+ tokenizer = AutoTokenizer.from_pretrained("Josephgflowers/TinyLlama-3T-Cinder-v1.3")
26
+ model = AutoModelForCausalLM.from_pretrained("Josephgflowers/TinyLlama-3T-Cinder-v1.3")
 
 
 
 
27
 
28
  # Load pre-trained Sentence Transformer model
29
  sbert_model_name = "paraphrase-MiniLM-L6-v2"