randomtable commited on
Commit
aff5c56
·
verified ·
1 Parent(s): abf6ee3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ def inference(img, version, scale):
101
  if img_mode == 'RGBA': # RGBA images should be saved in png format
102
  extension = 'png'
103
  else:
104
- extension = 'jpg'
105
  save_path = f'output/out.{extension}'
106
  cv2.imwrite(save_path, output)
107
 
 
101
  if img_mode == 'RGBA': # RGBA images should be saved in png format
102
  extension = 'png'
103
  else:
104
+ extension = 'png'
105
  save_path = f'output/out.{extension}'
106
  cv2.imwrite(save_path, output)
107