Update app.py

app.py

@@ -2,10 +2,15 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 
-# Function to create an image with random pixels
+# Function to create an image with exactly num_pixels random pixels
 def generate_random_image(num_pixels):
     width, height = 100, 100  # Set the dimensions of the image (adjust as needed)
-    image = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
+    image = np.zeros((height, width, 3), dtype=np.uint8)  # Create a black image
+    if num_pixels > 0:
+        random_coordinates = np.random.randint(0, width, (num_pixels, 2))
+        for coord in random_coordinates:
+            image[coord[0], coord[1]] = np.random.randint(0, 256, (3,), dtype=np.uint8)  # Set random pixel color
+
     image = Image.fromarray(image)
     return image
 
@@ -14,8 +19,8 @@ iface = gr.Interface(
     fn=generate_random_image,
     inputs="number",
     outputs="image",
-    title="Random Image Generator",
-    description="Enter the number of pixels to generate a random image.",
+    title="Random Pixel Image Generator",
+    description="Enter the number of random pixels to generate an image.",
 )
 
 iface.launch()