Update app.py

Changed the default values of the input style and content image

app.py

@@ -98,8 +98,14 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
 
 
 #Defining the predict function
-def style_transfer(cont_img:Image.Image=torch.zeros((1,128,128)).detach().numpy(),styl_img:Image.Image=torch.zeros((1,128,128)).detach().numpy()) -> Image.Image:
-  
+def style_transfer(cont_img:Image.Image=None,styl_img:Image.Image=None) -> Image.Image:
+
+  if cont_img is None or styl_img is None:
+        # Handle the default behavior when inputs are not provided
+        # For example, you can load default images or return an error message
+        # Return a default output image or error message
+        return default_output_image
+      
   #Start the timer
   start_time = time.time()