Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 import pickle
 import uvicorn
 
@@ -89,6 +89,6 @@ async def predict(data: Predict):
                 (os.path.basename(str(document.metadata["source"])), str(document.page_content))
             )
 
-        return jsonify(prompt_response_dict), 200
+        return jsonify(prompt_response_dict)
     else:
-        return "No user prompt received", 400
+        raise HTTPException(status_code=400, detail="Prompt Incorrect")