Update app.py

app.py

@@ -37,8 +37,7 @@ async def check_generation_status(job_id: str):
         return JSONResponse(content={"status": "not_found"}, status_code=404)
 
 
-# Mount the Gradio Blocks demo interface as WSGI
-api_app.mount("/", WSGIMiddleware(demo))
+
 
 # Install additional dependencies
 try:
@@ -757,6 +756,9 @@ except Exception as e:
     logger.error(f"Failed to initialize Gradio Blocks interface: {str(e)}")
     raise
 
+# Mount the Gradio Blocks demo interface as WSGI
+api_app.mount("/", WSGIMiddleware(demo))
+
 if __name__ == "__main__":
     try:
         logger.info("Launching Gradio application")