Update app.py

app.py

@@ -40,6 +40,9 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         generator=generator
     ).images[0]
 
+    # Ensure image is in uint8 format
+    image = (255 * np.clip(image, 0, 1)).astype(np.uint8)
+
     # Convert the image to PIL format for overlaying the watermark
     pil_image = Image.fromarray(image)
 
@@ -60,6 +63,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
 
 
 
+
 examples = [
     "Sunset over the Atlas Mountains",
     "Traditional Amazigh jewelry under the moonlight",