Update app.py

app.py

@@ -16,9 +16,8 @@ def augment_images(image_file, num_duplicates):
         horizontal_flip=True,
         fill_mode='nearest')
     
-    # Convert to RGB and resize the image
-    img = Image.open(image_file).convert('RGB')
-    img = img.resize((256, 256))
+    img = Image.open(image_file).convert('RGB')  # Convert to RGB
+    img = img.resize((256, 256))  # Resize image
     x = img_to_array(img)  # Convert image to numpy array
     x = x.reshape((1,) + x.shape)  # Reshape for data generator
     
@@ -30,7 +29,7 @@ def augment_images(image_file, num_duplicates):
                 break
 
         # Zip the augmented images
-        zip_name = os.path.join(tempfile.gettempdir(), 'augmented_images.zip')
+        zip_name = tempfile.mktemp(suffix='.zip')
         with zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED) as zipf:
             for root, _, files in os.walk(temp_dir):
                 for file in files:
@@ -40,10 +39,7 @@ def augment_images(image_file, num_duplicates):
 
 iface = gr.Interface(
     fn=augment_images,
-    inputs=[
-        gr.Image(tool="editor", label="Upload Image"),
-        gr.Slider(minimum=1, maximum=20, default=5, label="Number of Augmented Samples")
-    ],
+    inputs=[gr.Image(shape=None, label="Upload Image", source="upload"), gr.Slider(minimum=1, maximum=20, default=5, label="Number of Augmented Samples")],
     outputs=gr.File(label="Download Augmented Images"),
     title="Image Augmentation App",
     description="Upload an image to generate augmented versions. Select the number of augmented duplicates you want for the image."