Update app.py

app.py

@@ -454,7 +454,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             output_type="pil",
             good_vae=good_vae,
         ):
-            print(f"Debug: Yielding image of type {type(img)}")  # Check the type of the image
+            print(f"Debug: Yielding image of type {type(img)}")  # Check image type
             if isinstance(img, float):
                 print("Error: A float was returned instead of an image.")  # Log if img is a float
                 raise ValueError("Expected an image, but got a float.")  # Raise error if a float is found
@@ -462,6 +462,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             final_image = img  # Update final_image with the current image
         return final_image
 
+
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
     print("Generating image from input...")
     pipe_i2i.to("cuda")
@@ -585,7 +586,6 @@ def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selecte
 
 run_lora.zerogpu = True
 
-
 def get_huggingface_safetensors(link):
     split_link = link.split("/")
     if len(split_link) == 4: