Update app.py

app.py

@@ -40,7 +40,7 @@ default_models = models[:num_models]
 MAX_SEED = 2**32-1
 
 def imgageHasUpdated(theImage):
-    print(theImage)
+    print(theImage.result)
 
 def extend_choices(choices):
     return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
@@ -90,8 +90,9 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         with lock:
             png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
             # lnk_output(value=png_path)
-            imgageHasUpdated(png_path)
+            
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
+            imgageHasUpdated(image)
         return image
     return None