Namitg02 commited on
Commit
b1318a8
·
verified ·
1 Parent(s): ee478d9

Update app.py

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