replace examples, support manual seed

app.py

@@ -1,11 +1,12 @@
 import os
 import shlex
 import subprocess
+
 import gradio as gr
+import numpy as np
 import spaces
 import torch
 from diffusers import DiffusionPipeline
-from gradio_client import Client, file
 
 subprocess.run(
     shlex.split(
@@ -34,7 +35,10 @@ splat_pipeline = DiffusionPipeline.from_pretrained(
 
 
 @spaces.GPU
-def run(input_image):
+def run(input_image, seed):
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
     input_image = input_image.astype("float32") / 255.0
     images = image_pipeline(
         "", input_image, guidance_scale=5, num_inference_steps=30, elevation=0
@@ -78,27 +82,28 @@ with block:
     with gr.Row(variant="panel"):
         with gr.Column(scale=1):
             input_image = gr.Image(label="image", type="numpy")
+            seed_input = gr.Number(label="seed", value=42)
             button_gen = gr.Button("Generate")
 
         with gr.Column(scale=1):
             output_splat = gr.Model3D(label="3D Gaussians")
 
         button_gen.click(
-            fn=run, inputs=[input_image], outputs=[output_splat]
+            fn=run, inputs=[input_image, seed_input], outputs=[output_splat]
         )
 
     gr.Examples(
         examples=[
-            "data_test/frog_sweater.jpg",
-            "data_test/bird.jpg",
-            "data_test/boy.jpg",
-            "data_test/cat_statue.jpg",
-            "data_test/dragontoy.jpg",
-            "data_test/gso_rabbit.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/a_cat_statue.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/a_baby_penguin.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/A_cartoon_house_with_red_roof.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/a_hat.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/an_antique_chest.jpg",
+            "https://huggingface.co/datasets/dylanebert/iso3d/resolve/main/jpg@512/metal.jpg",
         ],
         inputs=[input_image],
         outputs=[output_splat],
-        fn=lambda x: run(input_image=x),
+        fn=lambda x: run(input_image=x, seed=42),
         cache_examples=True,
         label="Image-to-3D Examples",
     )