modify output_factor

ominicontrol.py

@@ -106,9 +106,9 @@ def generate_image(
     }[ratio]
 
 
-    width, height = image.size
-    width = int(width * ratio)
-    height = int(height * ratio)
+    output_factor = max(width, height) / max(original_width, original_height)
+    width = int(original_width * output_factor)
+    height = int(original_height * output_factor)
 
 
     print(