Update app.py
Browse files
app.py
CHANGED
@@ -98,7 +98,7 @@ def inference(img, version, scale):
|
|
98 |
if img_mode == 'RGBA': # RGBA images should be saved in png format
|
99 |
extension = 'png'
|
100 |
else:
|
101 |
-
extension = '
|
102 |
save_path = f'output/out.{extension}'
|
103 |
cv2.imwrite(save_path, output)
|
104 |
|
|
|
98 |
if img_mode == 'RGBA': # RGBA images should be saved in png format
|
99 |
extension = 'png'
|
100 |
else:
|
101 |
+
extension = 'png'
|
102 |
save_path = f'output/out.{extension}'
|
103 |
cv2.imwrite(save_path, output)
|
104 |
|