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