Update app.py

app.py

@@ -3,32 +3,21 @@ import albumentations as A
 import cv2
 import numpy as np
 
-# Function for image augmentation
 def augment_image(image, flip=False, rotate=0, brightness=1.0, noise=0, elastic=False, alpha=1.0, sigma=50):
     image = np.array(image)
     
     aug_list = []
-    if flip:
-        aug_list.append(A.HorizontalFlip(p=1.0))
-    
-    if rotate:
-        aug_list.append(A.Rotate(limit=rotate, p=1.0))
-    
-    if brightness != 1.0:
-        aug_list.append(A.RandomBrightnessContrast(brightness_limit=(brightness-1, brightness-1), p=1.0))
-    
-    if noise > 0:
-        aug_list.append(A.GaussNoise(var_limit=(noise, noise), p=1.0))
-    
-    if elastic:
-        aug_list.append(A.ElasticTransform(alpha=alpha, sigma=sigma, alpha_affine=None, p=1.0))
+    if flip: aug_list.append(A.HorizontalFlip(p=1.0))
+    if rotate: aug_list.append(A.Rotate(limit=rotate, p=1.0))
+    if brightness != 1.0: aug_list.append(A.RandomBrightnessContrast(brightness_limit=(brightness-1, brightness-1), p=1.0))
+    if noise > 0: aug_list.append(A.GaussNoise(var_limit=(noise, noise), p=1.0))
+    if elastic: aug_list.append(A.ElasticTransform(alpha=alpha, sigma=sigma, alpha_affine=None, p=1.0))
     
     aug = A.Compose(aug_list)
     augmented = aug(image=image)
     
     return augmented["image"]
 
-# Gradio Interface
 def image_augmentor_interface(image, flip, rotate, brightness, noise, elastic, alpha, sigma):
     augmented_image = augment_image(image, flip, rotate, brightness, noise, elastic, alpha, sigma)
     return augmented_image
@@ -39,7 +28,7 @@ iface = gr.Interface(
         gr.Image(type="numpy"),
         gr.Checkbox(label="Flip Horizontally"),
         gr.Slider(0, 360, label="Rotate Degrees"),
-        gr.Slider(0.5, 1.5, step=0.1, label="Brightness Adjustment"),
+        gr.Slider(0.5, 1.5, step=0.1, value=1, label="Brightness Adjustment"),
         gr.Slider(0, 100, step=1, label="Noise Scale"),
         gr.Checkbox(label="Elastic Distortion"),
         gr.Slider(1.0, 200.0, step=1.0, label="Elastic Alpha (distortion intensity)"),