1inkusFace commited on
Commit
f9a089d
·
verified ·
1 Parent(s): 52b4a6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,6 +10,7 @@ import numpy as np
10
 
11
  # Removed environment-specific lines
12
  from diffusers.utils import export_to_video
 
13
 
14
  import torch
15
  import logging
@@ -263,7 +264,7 @@ def generate_video(prompt: str, seed: int, image: str = None) -> tuple[str, dict
263
  model_id = "Skywork/SkyReels-V1-Hunyuan-I2V"
264
  kwargs = {
265
  "prompt": prompt,
266
- "image": Image.open(image),
267
  "height": 512,
268
  "width": 512,
269
  "num_frames": 97,
 
10
 
11
  # Removed environment-specific lines
12
  from diffusers.utils import export_to_video
13
+ from diffusers.utils import load_image
14
 
15
  import torch
16
  import logging
 
264
  model_id = "Skywork/SkyReels-V1-Hunyuan-I2V"
265
  kwargs = {
266
  "prompt": prompt,
267
+ "image": load_image(image),
268
  "height": 512,
269
  "width": 512,
270
  "num_frames": 97,