Namitg02 commited on
Commit
fd9f388
·
verified ·
1 Parent(s): 5347e77

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 = "deepset/roberta-base-squad2"
61
  tokenizer = AutoTokenizer.from_pretrained(llm_model)
62
  model = AutoModelForCausalLM.from_pretrained(llm_model)
63
- pipe = pipeline(model = llm_model, tokenizer = tokenizer,trust_remote_code=True, task = "question-answering", temperature=0.2)
64
 
65
  #question = "How can I reverse diabetes?"
66
  #docs1 = retriever.invoke(question)
 
60
  llm_model = "deepset/roberta-base-squad2"
61
  tokenizer = AutoTokenizer.from_pretrained(llm_model)
62
  model = AutoModelForCausalLM.from_pretrained(llm_model)
63
+ pipe = pipeline(model = llm_model, tokenizer = tokenizer,trust_remote_code=True, task = "question-answering", temperature=0.5)
64
 
65
  #question = "How can I reverse diabetes?"
66
  #docs1 = retriever.invoke(question)