Update app.py
Browse files
app.py
CHANGED
@@ -92,8 +92,8 @@ from transformers import AutoTokenizer
|
|
92 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
93 |
#question = "How can I reverse diabetes?"
|
94 |
|
95 |
-
pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "text-generation", temperature=0.2)
|
96 |
-
|
97 |
print("check2")
|
98 |
qa = ConversationalRetrievalChain.from_llm(
|
99 |
pipe,
|
|
|
92 |
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
93 |
#question = "How can I reverse diabetes?"
|
94 |
|
95 |
+
#pipe = pipeline(model = llm_model, tokenizer = tokenizer, task = "text-generation", temperature=0.2)
|
96 |
+
load_qa_chain(llm=llm_model,tokenizer =tokenizer, chain_type="map_reduce")
|
97 |
print("check2")
|
98 |
qa = ConversationalRetrievalChain.from_llm(
|
99 |
pipe,
|