Spaces:

JunhaoZhuang
/

Cobra

Runtime error

JunhaoZhuang commited on Apr 16

Commit

d31b15a

verified ·

1 Parent(s): 6ca968b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -425,6 +425,7 @@ def colorize_image(input_style, extracted_line, reference_images, resolution, se
     global MultiResNetModel
     global cur_style
     if input_style != cur_style:
         change_ckpt(input_style)
     tar_width, tar_height = resolution

     global MultiResNetModel
     global cur_style
     if input_style != cur_style:
+        gr.Info("Loading the model...")
         change_ckpt(input_style)
     tar_width, tar_height = resolution