Spaces:
Running
on
Zero
Running
on
Zero
Update app_14B.py
Browse files- app_14B.py +4 -3
app_14B.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
import uuid
|
3 |
import torch
|
@@ -10,12 +11,12 @@ from diffusers import WanImageToVideoPipeline
|
|
10 |
from diffusers.utils import export_to_video
|
11 |
from huggingface_hub import upload_file
|
12 |
from PIL import Image
|
13 |
-
|
14 |
|
15 |
# ----------------- Setup -----------------
|
16 |
logging.basicConfig(level=logging.INFO)
|
17 |
|
18 |
-
HF_MODEL = "rahul7star/rahulAI"
|
19 |
dtype = torch.bfloat16
|
20 |
device = "cuda"
|
21 |
|
@@ -137,7 +138,7 @@ def generate_video(input_image, prompt, negative_prompt=default_negative_prompt,
|
|
137 |
export_to_video(output_frames_list, video_path, fps=16)
|
138 |
|
139 |
# Upload to HF
|
140 |
-
hf_folder = upscale_and_upload_4k(video_path, input_image, prompt)
|
141 |
|
142 |
return video_path, f"✅ Uploaded to HF: {hf_folder}"
|
143 |
|
|
|
1 |
+
import spaces
|
2 |
import os
|
3 |
import uuid
|
4 |
import torch
|
|
|
11 |
from diffusers.utils import export_to_video
|
12 |
from huggingface_hub import upload_file
|
13 |
from PIL import Image
|
14 |
+
|
15 |
|
16 |
# ----------------- Setup -----------------
|
17 |
logging.basicConfig(level=logging.INFO)
|
18 |
|
19 |
+
HF_MODEL = "rahul7star/rahulAI"
|
20 |
dtype = torch.bfloat16
|
21 |
device = "cuda"
|
22 |
|
|
|
138 |
export_to_video(output_frames_list, video_path, fps=16)
|
139 |
|
140 |
# Upload to HF
|
141 |
+
#hf_folder = upscale_and_upload_4k(video_path, input_image, prompt)
|
142 |
|
143 |
return video_path, f"✅ Uploaded to HF: {hf_folder}"
|
144 |
|