rasyosef commited on
Commit
a65c505
1 Parent(s): e832688

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -81,6 +81,7 @@ def generate(question, answer, text_file, max_new_tokens):
81
  pad_token_id=tokenizer.eos_token_id, eos_token_id=tokenizer.eos_token_id,
82
  device_map="cpu", streamer=streamer
83
  )
 
84
  hf_model = HuggingFacePipeline(pipeline=phi2_pipeline)
85
  qa_chain = get_retrieval_qa_chain(text_file, hf_model)
86
 
 
81
  pad_token_id=tokenizer.eos_token_id, eos_token_id=tokenizer.eos_token_id,
82
  device_map="cpu", streamer=streamer
83
  )
84
+
85
  hf_model = HuggingFacePipeline(pipeline=phi2_pipeline)
86
  qa_chain = get_retrieval_qa_chain(text_file, hf_model)
87