Update app.py

app.py

@@ -75,7 +75,7 @@ class LlavaCaptionProcessor:
 
 class MolmoCaptionProcessor:
     def __init__(self):
-        pretrained_model_name = 'allenai/Molmo-7B-O-0924'
+        pretrained_model_name = 'cyan2k/molmo-7B-D-bnb-4bit'
         self.processor = AutoProcessor.from_pretrained(
             pretrained_model_name,
             trust_remote_code=True,
@@ -712,7 +712,7 @@ with gr.Blocks(title="OneDiffusion Demo") as demo:
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description='Start the Gradio demo with specified captioner.')
-    parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='llava', help='Captioner to use: molmo, llava, disable.')
+    parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='molmo', help='Captioner to use: molmo, llava, disable.')
     args = parser.parse_args()
 
     # Initialize models with the specified captioner