Image-Caption-2-Shap-E

Paused

fffiloni commited on Jun 9, 2023

Commit

8ad7acc

1 Parent(s): 46b7d04

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -129,7 +129,7 @@ class Model:
             sigma_max=160,
             s_churn=0,
         )
-        return self.to_glb(latents[0])
     def run_image(self,
                   image_path: str,

             sigma_max=160,
             s_churn=0,
         )
+        return prompt, self.to_glb(latents[0])
     def run_image(self,
                   image_path: str,