Update app.py
Browse files
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.
|
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)
|