Update app.py

app.py

@@ -79,7 +79,7 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
     img_data_url = f"data:image/webp;base64,{img_base64}"
 
     latency = f"Latency: {(time.time()-start_time):.2f} seconds"
-    return img_data_url, generator.seed, latency
+    return img, generator.seed, latency
 
 # Example prompts
 examples = [