Spaces:
Paused
Paused
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 |
|