Update main.py

main.py

@@ -1,28 +1,25 @@
-from fastapi import FastAPI
-from models import sa_pipeline, query_index
-from pydantic import BaseModel
+from model import model_pipeline
 
-class QARequest(BaseModel):
-    question: str
+from typing import Union
 
-class QAResponse(BaseModel):
-    answer: str
+from fastapi import FastAPI, UploadFile
+import io
+from PIL import Image
 
-class SentimentRequest(BaseModel):
-    text: str
+app = FastAPI()
 
-class SentimentResponse(BaseModel):
-    label: str
-    score: float
 
-app = FastAPI()
+@app.get("/")
+def read_root():
+    return {"Hello": "World"}
 
-@app.post('/question-answering', response_model=QAResponse)
-def query(query: QARequest):
-    data = query.dict()
-    return {'answer': query_index(data['question'])}
 
-@app.post('/sentiment-analysis', response_model=SentimentResponse)
-def query(query: SentimentRequest):
-    data = query.dict()
-    return sa_pipeline(data['text'])[0]
+@app.post("/ask")
+def ask(text: str, image: UploadFile):
+    content = image.file.read()
+    
+    image = Image.open(io.BytesIO(content))
+    # image = Image.open(image.file)
+    
+    result = model_pipeline(text, image)
+    return {"answer": result}