Update app.py

app.py

@@ -139,7 +139,7 @@ def infer(
     print('-- filtered prompt --')
     print(enhanced_prompt)
     print('-- generating image --')
-      with torch.no_grad():
+    with torch.no_grad():
         sd_image = pipe(
             prompt=enhanced_prompt,  # This conversion is fine
             negative_prompt=negative_prompt,
@@ -168,37 +168,6 @@ css = """
 }
 """
 
-def repeat_infer(
-    prompt,
-    negative_prompt,
-    seed,
-    randomize_seed,
-    width,
-    height,
-    guidance_scale,
-    num_inference_steps,
-    num_iterations,  # New input for number of iterations
-):
-    i = 0
-    while i < num_iterations:
-        time.sleep(700)  # Wait for 10 minutes (600 seconds)
-        result, seed, image_path, enhanced_prompt = infer(
-            prompt,
-            negative_prompt,
-            seed,
-            randomize_seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-        )
-        
-        # Optionally, you can add logic here to process the results of each iteration
-        # For example, you could display the image, save it with a different name, etc.
-        i += 1
-    return result, seed, image_path, enhanced_prompt
-
-
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Text-to-Image StableDiffusion 3.5 Medium (with refine)")
@@ -222,9 +191,6 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter a negative prompt",
                 visible=False,
             )
-            num_iterations = gr.Number(
-                value=1000, 
-                label="Number of Iterations")
             seed = gr.Slider(
                 label="Seed",
                 minimum=0,