prithivMLmods commited on
Commit
045c0ad
·
verified ·
1 Parent(s): 5479f78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -382,12 +382,12 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
382
  image_submit.click(
383
  fn=generate_image,
384
  inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
385
- outputs=output
386
  )
387
  video_submit.click(
388
  fn=generate_video,
389
  inputs=[model_choice, video_query, video_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
390
- outputs=output
391
  )
392
 
393
  if __name__ == "__main__":
 
382
  image_submit.click(
383
  fn=generate_image,
384
  inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
385
+ outputs=[output, markdown_output]
386
  )
387
  video_submit.click(
388
  fn=generate_video,
389
  inputs=[model_choice, video_query, video_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
390
+ outputs=[output, markdown_output]
391
  )
392
 
393
  if __name__ == "__main__":