Namitg02 commited on
Commit
6484e7f
·
verified ·
1 Parent(s): f74d53d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,