Update app.py

app.py

@@ -114,7 +114,7 @@ normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],
                                  std=[0.26862954, 0.26130258, 0.27577711])
 lpips_model = lpips.LPIPS(net='vgg').to(device)
     
-def inference(text, image, skip_timesteps):
+def inference(text, init_image, skip_timesteps):
     all_frames = []
     prompts = [text]
     image_prompts = []
@@ -124,7 +124,10 @@ def inference(text, image, skip_timesteps):
     range_scale = 50            # Controls how far out of range RGB values are allowed to be.
     cutn = 16
     n_batches = 1
-    init_image = image.name   # This can be an URL or Colab local path and must be in quotes.
+    if init_image:
+        init_image = init_image.name
+    else:
+        init_image = None   # This can be an URL or Colab local path and must be in quotes.
     skip_timesteps = skip_timesteps  # This needs to be between approx. 200 and 500 when using an init image.
                         # Higher values make the output look more like the init.
     init_scale = 0      # This enhances the effect of the init image, a good value is 1000.