SamarthPujari commited on
Commit
f677e55
·
verified ·
1 Parent(s): f545f39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
63
  qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
64
 
65
  @Tool
66
- def document_qna_tool_fn(pdf_path: str, question: str) -> str:
67
  import traceback
68
 
69
  try:
 
63
  qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
64
 
65
  @Tool
66
+ def document_qna_tool(pdf_path: str, question: str) -> str:
67
  import traceback
68
 
69
  try: