Spaces:

SMD00
/

Image_Colorization

Runtime error

SMD00 commited on Jun 15, 2023

Commit

38e90aa

1 Parent(s): 3a3008e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -369,6 +369,7 @@ def main_func(filepath):
         model.setup_input(data)
         model.optimize()
         img=visualize(model, data, a)
     return (size_text,img)
 title = "Image Colorization"

         model.setup_input(data)
         model.optimize()
         img=visualize(model, data, a)
+        img=img.resize(a)
     return (size_text,img)
 title = "Image Colorization"