Update main.py

main.py

@@ -1,4 +1,6 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+
 from pydantic import BaseModel
 import pickle
 import uvicorn
@@ -9,7 +11,6 @@ import shutil
 import subprocess
 
 import torch
-from flask import Flask, jsonify, request, render_template
 from langchain.chains import RetrievalQA
 from langchain.embeddings import HuggingFaceInstructEmbeddings
 
@@ -90,6 +91,6 @@ async def predict(data: Predict):
                 (os.path.basename(str(document.metadata["source"])), str(document.page_content))
             )
 
-        return jsonify(prompt_response_dict)
+        return prompt_response_dict
     else:
         raise HTTPException(status_code=400, detail="Prompt Incorrect")