Update app.py

app.py

@@ -175,7 +175,7 @@ def text_to_image(prompt, show_thinking=False, cfg_text_scale=4.0, cfg_interval=
         image_shapes=image_shapes,
     )
 
-    result = {"text": ""}
+    result = {"text": None, "image": None}
     # Call inferencer with or without think parameter based on user choice
     for i in inferencer(text=prompt, think=show_thinking, **inference_hyper):
         if type(i) == str:
@@ -205,7 +205,7 @@ def image_understanding(image: Image.Image, prompt: str, show_thinking=False,
         max_think_token_n=max_new_tokens, # Set max_length
     )
     
-    result = {"text": ""}
+    result = {"text": None, "image": None}
     # Use show_thinking parameter to control thinking process
     for i in inferencer(image=image, text=prompt, think=show_thinking, 
                         understanding_output=True, **inference_hyper):
@@ -249,7 +249,7 @@ def edit_image(image: Image.Image, prompt: str, show_thinking=False, cfg_text_sc
     )
     
     # Include thinking parameter based on user choice
-    result = {"text": ""}
+    result = {"text": None, "image": None}
     for i in inferencer(image=image, text=prompt, think=show_thinking, **inference_hyper):
         if type(i) == str:
             result["text"] += i