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 type of each image
+            print(f"Debug: Yielding image of type {type(img)}")  # Check the type of the image
             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
@@ -463,6 +463,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
         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")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     image_input = load_image(image_input_path)
@@ -478,8 +479,12 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
         joint_attention_kwargs={"scale": 1.0},
         output_type="pil",
     ).images[0]
+    if isinstance(final_image, float):
+        print("Error: Expected an image but got a float.")
+        raise ValueError("Expected an image, but got a float.")
     return final_image
 
+
 @spaces.GPU(duration=75)
 def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, randomize_seed, seed, width, height, loras_state, image_input=None, progress=gr.Progress(track_tqdm=True)):
     print("run_lora function called.")  # Debugging statement
@@ -580,6 +585,7 @@ 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: