Set 9b as base model

app_dialogue.py

@@ -24,17 +24,17 @@ from transformers import AutoProcessor
 
 
 MODELS = [
-    # "HuggingFaceM4/idefics-9b-instruct",
-    "HuggingFaceM4/idefics-80b-instruct",
+    "HuggingFaceM4/idefics-9b-instruct",
+    # "HuggingFaceM4/idefics-80b-instruct",
 ]
 
 API_PATHS = {
     "HuggingFaceM4/idefics-9b-instruct": (
         "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-9b-instruct"
     ),
-    "HuggingFaceM4/idefics-80b-instruct": (
-        "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-80b-instruct"
-    ),
+    # "HuggingFaceM4/idefics-80b-instruct": (
+    #     "https://api-inference.huggingface.co/models/HuggingFaceM4/idefics-80b-instruct"
+    # ),
 }
 
 SYSTEM_PROMPT = [
@@ -690,7 +690,7 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
     with gr.Row(elem_id="model_selector_row"):
         model_selector = gr.Dropdown(
             choices=MODELS,
-            value="HuggingFaceM4/idefics-80b-instruct",
+            value="HuggingFaceM4/idefics-9b-instruct",
             interactive=True,
             show_label=False,
             container=False,