fixed bugs

check_app.py

@@ -116,10 +116,14 @@ MODEL_CONFIGS = {
     }
 }
 
-def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_scale=None, seed=None,negative_prompt,  randomize_seed, width, height, guidance_scale, num_inference_steps,  progress=gr.Progress(track_tqdm=True)):
+def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_scale=3.5, seed=None,negative_prompt=None,  randomize_seed=None, width=1024, height=1024, num_inference_steps=40,  progress=gr.Progress(track_tqdm=True)):
     generator = None
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
     if seed is not None:
         generator = torch.Generator("cuda").manual_seed(seed)
+    else:
+        generator = torch.Generator("cuda")
 
     def callback(pipe, step_index, timestep, callback_kwargs):
         print(f" callback => {step_index}, {timestep}")
@@ -160,7 +164,7 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
     # Generate image
     image = pipe(**common_args).images[0]
 
-    return image
+    return seed, image
 
 @spaces.GPU(duration=170)
 def create_pipeline_logic(prompt_text, model_name, negative_prompt="",  seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=40,):
@@ -184,10 +188,10 @@ def create_pipeline_logic(prompt_text, model_name, negative_prompt="",  seed=42,
     else:
         pipe = b_pipe
         
-    image = generate_image_with_progress(
-        model_name,pipe, prompt_text, num_steps=num_inference_steps, guidance_scale=guidance_scale, seed=seed,negative_prompt,  randomize_seed, width, height, guidance_scale, num_inference_steps, progress=progress
+    gen_seed,image = generate_image_with_progress(
+        model_name,pipe, prompt_text, num_steps=num_inference_steps, guidance_scale=guidance_scale, seed=seed,negative_prompt = negative_prompt,  randomize_seed = randomize_seed, width = width, height = height, progress=progress
     )
-    return f"Seed: {seed}", image
+    return f"Seed: {gen_seed}", image
 
 def main():
     with gr.Blocks() as app: