Spaces:

VAST-AI
/

DetailGen3D

Running on Zero

Aluren commited on Apr 16

Commit

cb9f286

verified ·

1 Parent(s): 7286300

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,11 +157,11 @@ def run_refinement(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    print("rgb_image", rgb_image)
     # print("mesh", rgb_image)
     if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
-        rgb_image = rgb_image["image"]
     scene = run_detailgen3d(
         pipeline,

     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+    # print("rgb_image", rgb_image)
     # print("mesh", rgb_image)
     if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
+        rgb_image = Image.open(rgb_image["image"]).convert("RGB")
     scene = run_detailgen3d(
         pipeline,