Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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"
|