SMD00 commited on
Commit
74e320f
1 Parent(s): c46026d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -369,9 +369,9 @@ def main_func(filepath):
369
  model.setup_input(data)
370
  model.optimize()
371
  img=visualize(model, data, a)
372
- img2=Image.fromarray(np.resize(img,a))
373
- img3=cv.resize(np.uint8(img), a, interpolation=cv.INTER_CUBIC)
374
- img4=cv.resize(np.uint8(img), a, interpolation=cv.INTER_LINEAR)
375
  return (size_text,img,img2,img3,img4)
376
 
377
  title = "Image Colorization"
 
369
  model.setup_input(data)
370
  model.optimize()
371
  img=visualize(model, data, a)
372
+ img2=np.resize(img,a)
373
+ img3=Image.fromarray(cv.resize(np.uint8(img), a, interpolation=cv.INTER_CUBIC))
374
+ img4=Image.fromarray(cv.resize(np.uint8(img), a, interpolation=cv.INTER_LINEAR))
375
  return (size_text,img,img2,img3,img4)
376
 
377
  title = "Image Colorization"