Update app_v3.py

app_v3.py

@@ -22,7 +22,7 @@ from typing import Generator
 # from peft import PeftModel, PeftConfig
 import gradio as gr
 from huggingface_hub import CommitScheduler, HfApi, logging
-from debug import log_params, scheduler
+from debug import log_params, scheduler, save_image
 logging.set_verbosity_debug()
 
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
@@ -105,7 +105,7 @@ def generate_caption(control_image):
         return None, None
     
     # Generate a detailed caption
-    mcaption = model.caption(control_image, length="normal")
+    mcaption = model.caption(control_image, length="short")
     detailed_caption = mcaption["caption"]
     print(f"Detailed caption: {detailed_caption}")
     
@@ -152,7 +152,12 @@ def process_image(control_image, user_prompt, system_prompt, scale, steps,
             seed=seed,
             guidance_end=guidance_end
         )
-        image.save("/tmp/" + str(seed) + "output.png")
+        
+        try:
+            debug_img = Image.open(image.save("/tmp/" + str(seed) + "output.png"))
+            save_image("/tmp/" + str(seed) + "output.png", debug_img)
+        except Exception as e:
+            print("Error 160: " + str(e))
         log_params(final_prompt, scale, steps, controlnet_conditioning_scale, guidance_scale, seed, guidance_end, control_image, image)
         yield f"Completed! Used prompt: {final_prompt}", image, final_prompt
     except Exception as e: