Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,9 @@ model = AutoModelForCausalLM.from_pretrained(llm_model)
|
|
| 63 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer,trust_remote_code=True, task = "question-answering", temperature=0.2)
|
| 64 |
|
| 65 |
question = "How can I reverse diabetes?"
|
| 66 |
-
docs1 = retriever.invoke(question)
|
|
|
|
|
|
|
| 67 |
print(docs1[0].page_content)
|
| 68 |
#print(docs1)[0]['generated_text'][-1]
|
| 69 |
|
|
|
|
| 63 |
pipe = pipeline(model = llm_model, tokenizer = tokenizer,trust_remote_code=True, task = "question-answering", temperature=0.2)
|
| 64 |
|
| 65 |
question = "How can I reverse diabetes?"
|
| 66 |
+
#docs1 = retriever.invoke(question)
|
| 67 |
+
docs1 = retriever.invoke(get_relevant_documents)
|
| 68 |
+
|
| 69 |
print(docs1[0].page_content)
|
| 70 |
#print(docs1)[0]['generated_text'][-1]
|
| 71 |
|