lehduong commited on
Commit
69bafd6
1 Parent(s): 46341b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -709,10 +709,10 @@ with gr.Blocks(title="OneDiffusion Demo") as demo:
709
 
710
  if __name__ == "__main__":
711
  parser = argparse.ArgumentParser(description='Start the Gradio demo with specified captioner.')
712
- parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='molmo', help='Captioner to use: molmo, llava, disable.')
713
  args = parser.parse_args()
714
 
715
  # Initialize models with the specified captioner
716
  pipeline, captioner = initialize_models(args.captioner)
717
 
718
- demo.launch(debug=True, ssr_mode=False)
 
709
 
710
  if __name__ == "__main__":
711
  parser = argparse.ArgumentParser(description='Start the Gradio demo with specified captioner.')
712
+ parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='llava', help='Captioner to use: molmo, llava, disable.')
713
  args = parser.parse_args()
714
 
715
  # Initialize models with the specified captioner
716
  pipeline, captioner = initialize_models(args.captioner)
717
 
718
+ demo.launch(debug=True)