1inkusFace commited on
Commit
a40e1b1
·
verified ·
1 Parent(s): 0e0805e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,13 +9,13 @@ from PIL import Image
9
  import numpy as np # Import NumPy
10
 
11
 
12
- # subprocess.run(['sh', './sky.sh']) # Keep if needed
13
  sys.path.append("./SkyReels-V1")
14
 
15
  # Corrected Relative Imports
16
- from SkyReels-V1.skyreelsinfer import TaskType
17
- from SkyReels-V1.skyreelsinfer.offload import OffloadConfig
18
- from SkyReels-V1.skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer # Import the class
19
  from diffusers.utils import export_to_video
20
  import torch
21
  import logging
 
9
  import numpy as np # Import NumPy
10
 
11
 
12
+ subprocess.run(['sh', './sky.sh']) # Keep if needed
13
  sys.path.append("./SkyReels-V1")
14
 
15
  # Corrected Relative Imports
16
+ from skyreelsinfer import TaskType
17
+ from skyreelsinfer.offload import OffloadConfig
18
+ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer # Import the class
19
  from diffusers.utils import export_to_video
20
  import torch
21
  import logging