Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -113,7 +113,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
|
|
113 |
segment_timesteps = torch.from_numpy(timesteps_split_np[0]).to("cuda")
|
114 |
num_channels_latents = pipe.transformer.config.in_channels
|
115 |
num_channels_latents = int(num_channels_latents / 2)
|
116 |
-
image = Image.open(image
|
117 |
image.resize((size,size), Image.LANCZOS)
|
118 |
with torch.no_grad():
|
119 |
image = pipe.video_processor.preprocess(image, height=size, width=size).to(
|
|
|
113 |
segment_timesteps = torch.from_numpy(timesteps_split_np[0]).to("cuda")
|
114 |
num_channels_latents = pipe.transformer.config.in_channels
|
115 |
num_channels_latents = int(num_channels_latents / 2)
|
116 |
+
image = Image.open(image).convert('RGB')
|
117 |
image.resize((size,size), Image.LANCZOS)
|
118 |
with torch.no_grad():
|
119 |
image = pipe.video_processor.preprocess(image, height=size, width=size).to(
|