Update

README.md

@@ -4,7 +4,7 @@ emoji: 🧢
 colorFrom: yellow
 colorTo: blue
 sdk: gradio
-sdk_version: 3.36.1
+sdk_version: 3.38.0
 python_version: 3.10.11
 app_file: app.py
 pinned: false

model.py

@@ -12,12 +12,12 @@ class Model:
     def __init__(self):
         self.device = torch.device(
             'cuda' if torch.cuda.is_available() else 'cpu')
-        self.pipe = ShapEPipeline.from_pretrained('YiYiXu/shap-e',
+        self.pipe = ShapEPipeline.from_pretrained('openai/shap-e',
                                                   torch_dtype=torch.float16)
         self.pipe.to(self.device)
 
         self.pipe_img = ShapEImg2ImgPipeline.from_pretrained(
-            'YiYiXu/shap-e-img2img', torch_dtype=torch.float16)
+            'openai/shap-e-img2img', torch_dtype=torch.float16)
         self.pipe_img.to(self.device)
 
     def to_glb(self, ply_path: str) -> str:

requirements.txt

@@ -1,5 +1,5 @@
-git+https://github.com/huggingface/diffusers@shap-ee-mesh
-gradio==3.36.1
+git+https://github.com/huggingface/diffusers@09fab56
+gradio==3.38.0
 torch==2.0.1
 torchvision==0.15.2
 trimesh==3.22.3