Update app.py

app.py

@@ -457,7 +457,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             #print("Image generated successfully.")  # Debugging statement
             yield img
             final_image = img  # Update final_image with the current image
-    return final_image
+        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)):
@@ -529,6 +529,8 @@ def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selecte
     try:
         if image_input is not None:
             final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, seed)
+            print("Debug: generate_image_to_image output type:", type(final_image))  # Debugging
+            print("Debug: generate_image_to_image output value:", final_image)  # Debugging
             yield final_image, seed, gr.update(visible=False)
         else:
             image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
@@ -536,6 +538,8 @@ def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selecte
             final_image = None
             step_counter = 0
             for image in image_generator:
+                print("Debug: generate_image yielded type:", type(image))  # Debugging
+                print("Debug: generate_image yielded value:", image)  # Debugging
                 step_counter += 1
                 final_image = image
                 print(f"Yielding image {step_counter}/{steps}")  # Debugging statement
@@ -543,6 +547,10 @@ def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selecte
                 yield image, seed, gr.update(value=progress_bar, visible=True)
             if final_image is None:
                 print("No final image generated.")  # Debugging statement
+            else:
+                print("Debug: final_image type:", type(final_image))  # Debugging
+                print("Debug: final_image value:", final_image)  # Debugging
+                
             yield final_image, seed, gr.update(value=progress_bar, visible=False)
 ##            
         yield None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None  # Placeholder for missing outputs