Update python version

README.md

@@ -4,6 +4,7 @@ emoji: 🚀
 colorFrom: yellow
 colorTo: green
 sdk: gradio
+python_version: 3.10
 sdk_version: 3.28.2
 app_file: app.py
 pinned: true

app.py

@@ -69,7 +69,8 @@ def generate_3D(input, grid_size=64):
     diffusion = diffusion_from_config(load_config('diffusion'))
     batch_size = 4
 
-    print(type(input))
+    if isinstance(input, np.ndarray):
+        input = Image.fromarray(input)
 
     if isinstance(input, Image.Image):
         input = prepare_img(input)
@@ -114,7 +115,7 @@ def generate_3D(input, grid_size=64):
 
     tm=rm.tri_mesh()
 
-    with open("output/mesh.ply",'wb') as f:
+    with open("/tmp/mesh.ply",'wb') as f:
         tm.write_ply(f)
 
 
@@ -129,11 +130,10 @@ def generate_3D(input, grid_size=64):
     images = decode_latent_images(xm, latents[0], cameras, rendering_mode=render_mode)
 
     # Convert images to gif
-    os.makedirs('output', exist_ok=True)
-    images[0].save(f'/tmp/mesh.mp4', save_all=True, append_images=images[1:], duration=100, loop=0)
+    images[0].save(f'/tmp/mesh.gif', save_all=True, append_images=images[1:], duration=100, loop=0)
 
 
-    return ply_to_glb('/tmp/mesh.ply', '/tmp/mesh.glb'), '/tmp/mesh.mp4', gr.update(value=['/tmp/mesh.glb', '/tmp/mesh.ply'], visible=True)
+    return ply_to_glb('/tmp/mesh.ply', '/tmp/mesh.glb'), '/tmp/mesh.gif', gr.update(value=['/tmp/mesh.glb', '/tmp/mesh.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -198,6 +198,8 @@ with block:
                     <a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/MirageML/point-e?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>
                 </div>
                 <br />
+                If the Queue is Too Long, <a href="https://app.mirageml.com/generate" style="text-decoration: underline" target="_blank">Try it on Mirage</a>!
+                <br />
                 <p style="margin-bottom: 10px; font-size: 94%">
                 Generate 3D Assets in 1 minute with a prompt or image!
                 Based on the <a href="https://github.com/openai/shap-e">Shap-E</a> implementation
@@ -220,6 +222,8 @@ with block:
                         <a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/MirageML/point-e?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>
                     </div>
                     <br />
+                    If the Queue is Too Long, <a href="https://app.mirageml.com/generate" style="text-decoration: underline" target="_blank">Try it on Mirage</a>!
+                    <br />
                     <p style="margin-bottom: 10px; font-size: 94%">
                     Generate 3D Assets in 1 minute with a prompt or image!
                     Based on the <a href="https://github.com/openai/shap-e">Shap-E</a> implementation
@@ -261,6 +265,16 @@ with block:
             fn=generate_3D,
             cache_examples=True
         )
+        gr.Examples(
+            examples=[
+                ["a shark"],
+                ["an avocado"],
+            ],
+            inputs=[prompt],
+            outputs=[model_3d, model_gif, file_out],
+            fn=generate_3D,
+            cache_examples=True
+        )
 
     img_button.click(fn=generate_3D, inputs=[input_image], outputs=[model_3d, model_gif, file_out])
     text_button.click(fn=generate_3D, inputs=[prompt], outputs=[model_3d, model_gif, file_out])