Update app.py

app.py

@@ -59,6 +59,18 @@ def append_to_history(image, prompt, seed, width, height, guidance_scale, steps,
     if image is None:
         return history
     
+    # Convert numpy array to PIL Image if needed
+    from PIL import Image
+    import numpy as np
+    
+    if isinstance(image, np.ndarray):
+        # Convert from [0-255] to PIL Image
+        if image.dtype == np.uint8:
+            image = Image.fromarray(image)
+        # Convert from float [0-1] to PIL Image
+        else:
+            image = Image.fromarray((image * 255).astype(np.uint8))
+    
     # Convert final image to bytes
     buffered = BytesIO()
     image.save(buffered, format="PNG")