Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ question = "How can I reverse Diabetes?"
|
|
57 |
from transformers import AutoTokenizer
|
58 |
from transformers import AutoModelForCausalLM
|
59 |
|
60 |
-
llm_model = "
|
61 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
62 |
model = AutoModelForCausalLM.from_pretrained(llm_model)
|
63 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "text-generation", temperature=0.5)
|
|
|
57 |
from transformers import AutoTokenizer
|
58 |
from transformers import AutoModelForCausalLM
|
59 |
|
60 |
+
llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
61 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
62 |
model = AutoModelForCausalLM.from_pretrained(llm_model)
|
63 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "text-generation", temperature=0.5)
|