rahul7star commited on
Commit
88742fc
·
verified ·
1 Parent(s): 3ada3cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -337,6 +337,9 @@ def generate_video(
337
  progress=gr.Progress(track_tqdm=True),
338
  ):
339
  print("Prompt:", prompt)
 
 
 
340
 
341
  num_frames = np.clip(int(round(duration_seconds * FIXED_FPS)), MIN_FRAMES_MODEL, MAX_FRAMES_MODEL)
342
  current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
 
337
  progress=gr.Progress(track_tqdm=True),
338
  ):
339
  print("Prompt:", prompt)
340
+ if "child" in prompt.lower():
341
+ print("Found 'child' in prompt. Exiting loop.")
342
+ break
343
 
344
  num_frames = np.clip(int(round(duration_seconds * FIXED_FPS)), MIN_FRAMES_MODEL, MAX_FRAMES_MODEL)
345
  current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)