rahul7star commited on
Commit
8565695
·
verified ·
1 Parent(s): 9071d89

Update app_torch.py

Browse files
Files changed (1) hide show
  1. app_torch.py +5 -2
app_torch.py CHANGED
@@ -15,6 +15,7 @@ from huggingface_hub import hf_hub_download
15
  from PIL import Image
16
  import spaces
17
  import time
 
18
  # ---------------------------
19
  # Environment flags (reduce fusion/compilation) — set early
20
  # ---------------------------
@@ -216,7 +217,9 @@ def generate_output(prompt, mode, duration, width, height, steps, guidance, sche
216
 
217
 
218
 
219
- out_name = f"/tmp/{int(time.time())}.{'mp4' if mode == 'video' else 'png'}"
 
 
220
 
221
 
222
  if mode == "image":
@@ -252,7 +255,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Kandinsky 5.0 T2V (robust load)")
252
  duration = gr.Slider(1, 10, step=1, value=5, label="Duration (s)")
253
  width = gr.Radio([512, 768], value=768, label="Width")
254
  height = gr.Radio([512, 768], value=512, label="Height")
255
- steps = gr.Slider(4, 50, step=1, value=25, label="Sampling Steps")
256
  guidance = gr.Slider(0.0, 20.0, step=0.5, value=8.0, label="Guidance Weight")
257
  scheduler = gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Scheduler Scale")
258
  btn = gr.Button("Generate", variant="primary")
 
15
  from PIL import Image
16
  import spaces
17
  import time
18
+ import time, random
19
  # ---------------------------
20
  # Environment flags (reduce fusion/compilation) — set early
21
  # ---------------------------
 
217
 
218
 
219
 
220
+
221
+ out_name = f"/tmp/{int(time.time())}_{random.randint(100,999)}.{'mp4' if mode == 'video' else 'png'}"
222
+
223
 
224
 
225
  if mode == "image":
 
255
  duration = gr.Slider(1, 10, step=1, value=5, label="Duration (s)")
256
  width = gr.Radio([512, 768], value=768, label="Width")
257
  height = gr.Radio([512, 768], value=512, label="Height")
258
+ steps = gr.Slider(4, 50, step=1, value=10, label="Sampling Steps")
259
  guidance = gr.Slider(0.0, 20.0, step=0.5, value=8.0, label="Guidance Weight")
260
  scheduler = gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Scheduler Scale")
261
  btn = gr.Button("Generate", variant="primary")