New UI

app.py

@@ -1,7 +1,33 @@
 import gradio as gr
+import torch
+from diffusers import ShapEPipeline
+from diffusers.utils import export_to_gif
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load the ShapE model
+ckpt_id = "openai/shap-e"
+pipe = ShapEPipeline.from_pretrained(ckpt_id).to("cuda")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate_shap_e_gif(prompt):
+    guidance_scale = 15.0
+    images = pipe(
+        prompt,
+        guidance_scale=guidance_scale,
+        num_inference_steps=64,
+        size=256,
+    ).images
+
+    gif_path = export_to_gif(images, f"{prompt}_3d.gif")
+    return gif_path
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=generate_shap_e_gif,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter a prompt"),
+    outputs="file",
+    title="ShapE 3D GIF Generator",
+    description="Enter a prompt to generate a 3D GIF using the ShapE model."
+)
+
+# Run the app
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+diffusers==0.29.0.dev0
+gradio==4.36.0
+torch==2.3.1