Dewasheesh commited on
Commit
64ce359
·
verified ·
1 Parent(s): ad6e44b

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +6 -1
app/main.py CHANGED
@@ -1,6 +1,6 @@
1
  from pydantic import BaseModel
2
  from fastapi import FastAPI, UploadFile, File
3
- from app.pdf_handler import process_pdf, process_image_pdf
4
  from app.rag_pipeline import generate_answer, generate_answer_am
5
  from app.test import query_test
6
  from fastapi import FastAPI, File, UploadFile
@@ -37,6 +37,11 @@ async def upload_pdf_image(file: UploadFile = File(...)):
37
  return process_image_pdf(file)
38
 
39
 
 
 
 
 
 
40
  @app.post("/test/")
41
  async def chat(query: str, top_k: int = 3):
42
  return query_test(query)
 
1
  from pydantic import BaseModel
2
  from fastapi import FastAPI, UploadFile, File
3
+ from app.pdf_handler import process_pdf, process_image_pdf, process_image_pdf_dummy
4
  from app.rag_pipeline import generate_answer, generate_answer_am
5
  from app.test import query_test
6
  from fastapi import FastAPI, File, UploadFile
 
37
  return process_image_pdf(file)
38
 
39
 
40
+ @app.post("/upload/imgpdf/dummy/")
41
+ async def upload_pdf_image_dummy(f):
42
+ return process_image_pdf_dummy()
43
+
44
+
45
  @app.post("/test/")
46
  async def chat(query: str, top_k: int = 3):
47
  return query_test(query)