Update main.py

main.py

@@ -81,8 +81,16 @@ os.environ["GRADIO_ROOT_PATH"] = "https://m7n-zero-gpu-uvicorn-ssr-example.hf.sp
 
 # Mount Gradio app to FastAPI with SSR mode for Spaces
 app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True)
-app.mount("/_app/immutable/assets", StaticFiles(directory="assets"), name="immutable_assets")
+#app.mount("/_app/immutable/assets", StaticFiles(directory="assets"), name="immutable_assets")
+
+
+from pathlib import Path
+
+assets_dir = Path("./assets")
+if not assets_dir.exists():
+    assets_dir.mkdir(parents=True, exist_ok=True)  # Create the folder if it doesn't exist
 
+app.mount("/_app/immutable/assets", StaticFiles(directory=str(assets_dir)), name="immutable_assets")
 # Run server
 if __name__ == "__main__":
     # Set SSR mode for Spaces