Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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='
|
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
|
|
|
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)
|