Spaces:
Sleeping
Sleeping
Update app/main.py
Browse files- 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)
|