updated

app/main.py

@@ -1,14 +1,16 @@
 from fastapi import FastAPI
+import gradio as gr  
 from app.api.endpoints import staging
-from app.ui.gradio_ui import gradio_app 
+from app.ui.gradio_ui import gradio_app as gradio_blocks_ui
 
 app = FastAPI(
-    title="Auto Pick MobileCLIP API", 
-    description="An API to rank images using MobileCLIP2. Ideal for auto-picking the best photos for real estate, e-commerce, and more.", # <-- CHANGED
+    title="Auto Pick MobileCLIP API",
+    description="An API to rank images using MobileCLIP2. Ideal for auto-picking the best photos for real estate, e-commerce, and more.",
     version="1.0.0"
 )
 
 
 app.include_router(staging.router, prefix="/api/v1", tags=["Image Ranking"])
 
-app.mount("/", gradio_app) 
+
+app = gr.mount_gradio_app(app, gradio_blocks_ui, path="/")