Upload folder using huggingface_hub

index.py

@@ -17,31 +17,35 @@ load_dotenv(dotenv_path)
 REPLICATE_API_TOKEN = os.getenv('REPLICATE_API_TOKEN')
 
 def generate_pattern(image, prompt):
-    # Convert the numpy array to a PIL image
-    starter_image_pil = Image.fromarray(image.astype('uint8'))
-
-    # Resize the starter image if either dimension is larger than 768 pixels
-    if starter_image_pil.size[0] > 768 or starter_image_pil.size[1] > 768:
-        # Calculate the new size while maintaining the aspect ratio
-        if starter_image_pil.size[0] > starter_image_pil.size[1]:
-            # Width is larger than height
-            new_width = 768
-            new_height = int((768 / starter_image_pil.size[0]) * starter_image_pil.size[1])
-        else:
-            # Height is larger than width
-            new_height = 768
-            new_width = int((768 / starter_image_pil.size[1]) * starter_image_pil.size[0])
-        
-        # Resize the image
-        starter_image_pil = starter_image_pil.resize((new_width, new_height), Image.LANCZOS)
-
-
-    # Move the image horizontally and vertically by 50%
-    width, height = starter_image_pil.size
-    horizontal_shift = width // 2
-    vertical_shift = height // 2
-
-    transformed_image_pil = ImageChops.offset(starter_image_pil, horizontal_shift, vertical_shift)
+
+    if image is not None:
+        # Convert the numpy array to a PIL image
+        starter_image_pil = Image.fromarray(image.astype('uint8'))
+
+        # Resize the starter image if either dimension is larger than 768 pixels
+        if starter_image_pil.size[0] > 768 or starter_image_pil.size[1] > 768:
+            # Calculate the new size while maintaining the aspect ratio
+            if starter_image_pil.size[0] > starter_image_pil.size[1]:
+                # Width is larger than height
+                new_width = 768
+                new_height = int((768 / starter_image_pil.size[0]) * starter_image_pil.size[1])
+            else:
+                # Height is larger than width
+                new_height = 768
+                new_width = int((768 / starter_image_pil.size[1]) * starter_image_pil.size[0])
+            
+            # Resize the image
+            starter_image_pil = starter_image_pil.resize((new_width, new_height), Image.LANCZOS)
+
+
+        # Move the image horizontally and vertically by 50%
+        width, height = starter_image_pil.size
+        horizontal_shift = width // 2
+        vertical_shift = height // 2
+
+        transformed_image_pil = ImageChops.offset(starter_image_pil, horizontal_shift, vertical_shift)
+    else:
+        raise gr.Error(f"Please upload an image")        
 
 
     # Create a new image with black background and white cross