Spaces:

szymanowiczs
/

flash3d

Running on Zero

lyndonzheng commited on 18 days ago

Commit

bffa280

•

1 Parent(s): 8287c2f

update bugs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,9 +48,9 @@ def main():
             model.cfg.dataset.width, model.cfg.dataset.height = image.size
             model.set_backproject()
         else:
-            cfg.dataset.height, cfg.dataset.width = 256, 384
             image = TTF.resize(
-                image, (cfg.dataset.height, cfg.dataset.width),
                 interpolation=TT.InterpolationMode.BICUBIC
             )
         if padding:

             model.cfg.dataset.width, model.cfg.dataset.height = image.size
             model.set_backproject()
         else:
+            model.cfg.dataset.height, model.cfg.dataset.width = 256, 384
             image = TTF.resize(
+                image, (model.cfg.dataset.height, model.cfg.dataset.width),
                 interpolation=TT.InterpolationMode.BICUBIC
             )
         if padding: