update_caption

gradio_app.py

@@ -51,7 +51,7 @@ Generate consistent multi-view normals maps and color images.
 <div>
 The demo does not include the mesh reconstruction part, please visit <a href="https://github.com/xxlong0/Wonder3D/">our github repo</a> to get a textured mesh.
 </div>
-<span style="font-weight: bold; color: #d9534f;">- 2024.11.5 We shift our ckpt to the a more powerful model [Wonder3D_Plus] that supports both orthogonal and perspective camera settings and further improves generalizability.</span>
+<span style="font-weight: bold; color: #d9534f;">- 2024.12 We shift our ckpt to the a more powerful model [Wonder3D_Plus] that supports both orthogonal and perspective camera settings and further improves generalizability.</span>
 '''
 _GPU_ID = 0
 
@@ -61,8 +61,8 @@ if not hasattr(Image, 'Resampling'):
 
 
 def sam_init():
-    model = SamModel.from_pretrained("facebook/sam-vit-huge").to("cuda")
-    processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
+    model = SamModel.from_pretrained("facebook/sam-vit-large").to("cuda")
+    processor = SamProcessor.from_pretrained("facebook/sam-vit-large")
     return model, processor
 
 def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):