Update app.py

app.py

@@ -1,8 +1,22 @@
 import streamlit as st
-from transformers import pipeline
+import torch
+from diffusers import ShapEPipeline
+from diffusers.utils import export_to_gif
 
-pipe = pipeline("text-generation", model="stabilityai/stable-code-3b", trust_remote_code=True)
+
+ckpt_id = "openai/shap-e"
+pipe = ShapEPipeline.from_pretrained(repo).to("cuda")
+
+
+guidance_scale = 15.0
 
 field = st.text_area(label='Promt')
 if st.button('Generate', use_container_width=True):
-    st.json(pipe(field))
+    images = pipe(
+        field,
+        guidance_scale=guidance_scale,
+        num_inference_steps=64,
+        size=256,
+    ).images
+    for i in images:
+        st.image(i)