abidlabs HF staff commited on
Commit
639d6fb
1 Parent(s): 6ddca14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -10
app.py CHANGED
@@ -39,11 +39,10 @@ def to_video(frames: list[np.ndarray], fps: int) -> str:
39
  return out_file.name
40
 
41
 
42
- def generate(prompt: str, seed: int, num_frames: int,
43
- num_inference_steps: int) -> str:
44
- seed = int(seed)
45
- num_frames = int(num_frames)
46
- num_inference_steps = int(num_inference_steps)
47
  if seed == -1:
48
  seed = random.randint(0, 1000000)
49
  generator = torch.Generator().manual_seed(seed)
@@ -54,11 +53,6 @@ def generate(prompt: str, seed: int, num_frames: int,
54
  return to_video(frames, 8)
55
 
56
 
57
- examples = [
58
- ['An astronaut riding a horse.', 0, 16, 25],
59
- ['A panda eating bamboo on a rock.', 0, 16, 25],
60
- ['Spiderman is surfing.', 0, 16, 25],
61
- ]
62
 
63
  with gr.Blocks(css='style.css') as demo:
64
  gr.Markdown(DESCRIPTION)
 
39
  return out_file.name
40
 
41
 
42
+ def generate(prompt: str) -> str:
43
+ seed = int(0)
44
+ num_frames = int(16)
45
+ num_inference_steps = int(25)
 
46
  if seed == -1:
47
  seed = random.randint(0, 1000000)
48
  generator = torch.Generator().manual_seed(seed)
 
53
  return to_video(frames, 8)
54
 
55
 
 
 
 
 
 
56
 
57
  with gr.Blocks(css='style.css') as demo:
58
  gr.Markdown(DESCRIPTION)