rahul7star commited on
Commit
23b71a7
·
verified ·
1 Parent(s): 88415e0

Update app_14B.py

Browse files
Files changed (1) hide show
  1. 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
- import spaces
14
 
15
  # ----------------- Setup -----------------
16
  logging.basicConfig(level=logging.INFO)
17
 
18
- HF_MODEL = "rahul7star/rahulAI" # 👈 change to your repo
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