Update main.py

main.py

@@ -14,6 +14,15 @@ app = FastAPI()
 hf_token = os.environ.get('HF_TOKEN')
 client = Client("Ashrafb/moondream_captioning", hf_token=hf_token)
 
+# Function to make API prediction
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
+
+# FastAPI setup
+app = FastAPI()
+
 # Function to make API prediction
 def predict_image_answer(file_path, Question):
     result = client.predict(file_path, Question, api_name="/get_caption")
@@ -26,10 +35,13 @@ async def answer(image: UploadFile = File(...), Question: str = Form(...)):
         with tempfile.NamedTemporaryFile(delete=False) as temp_file:
             shutil.copyfileobj(image.file, temp_file)
             temp_file_path = temp_file.name
+        logging.debug("Received file: %s", temp_file_path)
         answer = predict_image_answer(temp_file_path, Question)
+        logging.debug("Prediction result: %s", answer)
         os.unlink(temp_file_path)
         return answer
     except Exception as e:
+        logging.error("An error occurred: %s", e)
         raise HTTPException(status_code=500, detail=str(e))
 
 # Serving static files
@@ -39,3 +51,4 @@ app.mount("/", StaticFiles(directory="static", html=True), name="static")
 @app.get("/")
 def index() -> FileResponse:
     return FileResponse(path="/app/static/index.html", media_type="text/html")
+