Update app.py

app.py

@@ -27,8 +27,17 @@ my_string = "\n".join(my_list)
 text2 = "The application focuses on neural style transfer, where the style from a style image is applied to a content image."
 
 def perform_neural_transfer(content_image_input, style_image_input):
+    if content_image_input is None:
+        # Handle the case when content image is not provided
+        return PIL.Image.fromarray(np.zeros((1, 1, 3), dtype=np.uint8))
+
     # Load content images
     content_image = content_image_input.astype(np.float32)[np.newaxis, ...] / 255.
+
+    if style_image_input is None:
+        # Handle the case when style image is not provided
+        return tensor_to_image(content_image)
+
     style_image = style_image_input.astype(np.float32)[np.newaxis, ...] / 255.
 
     # Apply neural style transfer
@@ -38,6 +47,7 @@ def perform_neural_transfer(content_image_input, style_image_input):
     return tensor_to_image(stylized_image)
 
 
+
 app_interface = gr.Interface(
     fn=perform_neural_transfer,
     inputs=[content_image_input, style_image_input],