Update app.py

app.py

@@ -2,6 +2,8 @@
 
 import gradio as gr
 import torch
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 
 from app_image_to_3d import create_demo as create_demo_image_to_3d
 from app_text_to_3d import create_demo as create_demo_text_to_3d
@@ -14,6 +16,18 @@ if not torch.cuda.is_available():
 
 model = Model()
 
+# Create a FastAPI app
+app = FastAPI()
+
+# Add CORS middleware to allow cross-origin requests
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
 with gr.Blocks(css_paths="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Tabs():
@@ -22,5 +36,9 @@ with gr.Blocks(css_paths="style.css") as demo:
         with gr.Tab(label="Image to 3D"):
             create_demo_image_to_3d(model)
 
+# Mount the Gradio app
+app = gr.mount_gradio_app(app, demo, path="/")
+
 if __name__ == "__main__":
+    # For local development
     demo.queue(max_size=10).launch()