RamAnanth1 commited on
Commit
3499e0e
β€’
1 Parent(s): 5d778f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -99,7 +99,8 @@ def get_video(prompt):
99
  samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
100
  sampler=ddim_sampler,
101
  )
102
- return "Hello " + prompt + "!!"
 
103
 
104
  prompt_inp = gr.Textbox(label = "Prompt")
105
  iface = gr.Interface(fn=get_video, inputs=[prompt_inp], outputs="text")
 
99
  samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
100
  sampler=ddim_sampler,
101
  )
102
+
103
+ return samples[0]
104
 
105
  prompt_inp = gr.Textbox(label = "Prompt")
106
  iface = gr.Interface(fn=get_video, inputs=[prompt_inp], outputs="text")