Update app.py

app.py

@@ -12,6 +12,7 @@ import numpy as np
 import gradio as gr
 import psutil
 import time
+import math
 
 SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None)
 TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None)
@@ -78,8 +79,10 @@ async def predict(init_image, prompt, strength, steps, seed=1231231):
         init_image = resize_crop(init_image)
         generator = torch.manual_seed(seed)
         last_time = time.time()
-        if steps == 1:
-            strength = 1.0
+    
+        if int(steps * strength) < 1:
+            steps = math.ceil(1 / max(0.10, strength))
+            
         results = i2i_pipe(
             prompt=prompt,
             image=init_image,