Namitg02 commited on
Commit
0beb22f
·
verified ·
1 Parent(s): c283f58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ from transformers import AutoModelForCausalLM
60
  llm_model = "microsoft/Phi-3-mini-4k-instruct"
61
  tokenizer = AutoTokenizer.from_pretrained(llm_model,trust_remote_code=True)
62
  model = AutoModelForCausalLM.from_pretrained(llm_model,trust_remote_code=True)
63
- pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "text-generation", temperature=0.2)
64
 
65
  # "microsoft/Phi-3-mini-4k-instruct"
66
  question = "How can I reverse diabetes?"
 
60
  llm_model = "microsoft/Phi-3-mini-4k-instruct"
61
  tokenizer = AutoTokenizer.from_pretrained(llm_model,trust_remote_code=True)
62
  model = AutoModelForCausalLM.from_pretrained(llm_model,trust_remote_code=True)
63
+ pipe = pipeline(model = llm_model, tokenizer = tokenizer,trust_remote_code=True, task = "text-generation", temperature=0.2)
64
 
65
  # "microsoft/Phi-3-mini-4k-instruct"
66
  question = "How can I reverse diabetes?"