Update app.py

app.py

@@ -206,19 +206,20 @@ def process_image(input_img):
     # print(input_glb)
     return input_glb, inference_glb
 
-gr.Markdown("""
-## Image to 3D in cluttered scene with [RaySt3R](https://rayst3r.github.io/)
-* Upload an image and click submit. 
-* We mask the input using [Rembg](https://github.com/danielgatis/rembg) and generate depth with [MoGe](https://github.com/microsoft/MoGe).
-* The input point cloud will be visible in 'input', the output is visible in the "RaySt3R" section.
-
-[Project Website](https://rayst3r.github.io/)
-""")
-demo = gr.Interface(
-    process_image,
-    gr.Image(),
-    [gr.Model3D(label="Input"), gr.Model3D(label="RaySt3R",)]
-)
+with gr.Blocks(delete_cache=(600, 600)) as demo:
+    gr.Markdown("""
+    ## Image to 3D in cluttered scene with [RaySt3R](https://rayst3r.github.io/)
+    * Upload an image and click submit. 
+    * We mask the input using [Rembg](https://github.com/danielgatis/rembg) and generate depth with [MoGe](https://github.com/microsoft/MoGe).
+    * The input point cloud will be visible in 'input', the output is visible in the "RaySt3R" section.
+    
+    [Project Website](https://rayst3r.github.io/)
+    """)
+    gr.Interface(
+        process_image,
+        gr.Image(),
+        [gr.Model3D(label="Input"), gr.Model3D(label="RaySt3R",)]
+    )
 
 if __name__ == "__main__":
     dummy_warmup()