xp3857 commited on
Commit
3ca5778
·
1 Parent(s): 5238c83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -136,7 +136,8 @@ def run(image,Restoration_mode, Colorizaition_mode):
136
  os.makedirs("Temp")
137
  os.makedirs("Temp/input")
138
  print(type(image))
139
- image=image.thumbnail((800,800))
 
140
  #image=np.array(image)
141
  image=cv2.imread(image)
142
  cv2.imwrite("Temp/input/input_img.png", image)
 
136
  os.makedirs("Temp")
137
  os.makedirs("Temp/input")
138
  print(type(image))
139
+ max_size=(800,800)
140
+ image=image.thumbnail(max_size)
141
  #image=np.array(image)
142
  image=cv2.imread(image)
143
  cv2.imwrite("Temp/input/input_img.png", image)