OmPrakashSingh1704 commited on
Commit
0bb12d1
1 Parent(s): 3039940
options/Video_model/Model.py CHANGED
@@ -3,6 +3,7 @@ from diffusers import StableVideoDiffusionPipeline
3
  from diffusers.utils import load_image, export_to_video
4
  from PIL import Image
5
  from huggingface_hub import login
 
6
  login(token=os.getenv("TOKEN"))
7
  # Check if CUDA (GPU) is available, otherwise use CPU
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
3
  from diffusers.utils import load_image, export_to_video
4
  from PIL import Image
5
  from huggingface_hub import login
6
+ import numpy as np
7
  login(token=os.getenv("TOKEN"))
8
  # Check if CUDA (GPU) is available, otherwise use CPU
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
options/Video_model/__pycache__/Model.cpython-310.pyc CHANGED
Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ