Update app.py
Browse files
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 = '
|
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 |
|