Namitg02 commited on
Commit
8c7f62e
·
verified ·
1 Parent(s): b317f13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "unsloth/Phi-3-mini-4k-instruct"
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)