Update app.py
Browse files
app.py
CHANGED
@@ -98,14 +98,10 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
|
|
98 |
print('Error', error)
|
99 |
print('If you encounter CUDA out of memory, try to set --tile with a smaller number.')
|
100 |
else:
|
101 |
-
|
102 |
-
if img_mode == 'RGBA': # RGBA images should be saved in png format
|
103 |
-
extension = 'jpg'
|
104 |
-
else:
|
105 |
-
extension = 'jpg'
|
106 |
|
107 |
out_filename = f"output_{rnd_string(8)}.{extension}"
|
108 |
-
cv2.imwrite(out_filename, output)
|
109 |
global last_file
|
110 |
last_file = out_filename
|
111 |
return out_filename
|
|
|
98 |
print('Error', error)
|
99 |
print('If you encounter CUDA out of memory, try to set --tile with a smaller number.')
|
100 |
else:
|
101 |
+
extension = 'jpg'
|
|
|
|
|
|
|
|
|
102 |
|
103 |
out_filename = f"output_{rnd_string(8)}.{extension}"
|
104 |
+
cv2.imwrite(out_filename, output, [int(cv2.IMWRITE_JPEG_QUALITY), 90]) # 90 là chất lượng JPEG (có thể chỉnh)
|
105 |
global last_file
|
106 |
last_file = out_filename
|
107 |
return out_filename
|