Update app.py

app.py

@@ -180,7 +180,7 @@ def text_to_image(prompt, show_thinking=False, cfg_text_scale=4.0, cfg_interval=
     for i in inferencer(text=prompt, think=show_thinking, **inference_hyper):
         if type(i) == str:
             result["text"] += i
-        elif type(i) == Image.Image:
+        else:
             result["image"] = i
 
         yield result["image"], result.get("text", None)
@@ -211,7 +211,7 @@ def image_understanding(image: Image.Image, prompt: str, show_thinking=False,
                         understanding_output=True, **inference_hyper):
         if type(i) == str:
             result["text"] += i
-        elif type(i) == Image.Image:
+        else:
             result["image"] = i
         yield result["text"]
 
@@ -253,7 +253,7 @@ def edit_image(image: Image.Image, prompt: str, show_thinking=False, cfg_text_sc
     for i in inferencer(image=image, text=prompt, think=show_thinking, **inference_hyper):
         if type(i) == str:
             result["text"] += i
-        elif type(i) == Image.Image:
+        else:
             result["image"] = i
 
         yield result["image"], result.get("text", "")