Update app_dialogue.py

app_dialogue.py

@@ -30,14 +30,17 @@ DEVICE = torch.device("cuda")
 MODELS = {
     "idefics2-8b-chatty": Idefics2ForConditionalGeneration.from_pretrained(
         "HuggingFaceM4/idefics2-8b-chatty",
+        # "Ali-C137/idefics2-8b-chatty-yalla",
         torch_dtype=torch.bfloat16,
         _attn_implementation="flash_attention_2",
     ).to(DEVICE),
 }
 PROCESSOR = AutoProcessor.from_pretrained(
     "HuggingFaceM4/idefics2-8b",
+    # "Ali-C137/idefics2-8b-chatty-yalla",
 )
 
+# Should change this section for the finetuned model
 SYSTEM_PROMPT = [
     {
         "role": "system",
@@ -84,89 +87,10 @@ EXAMPLES = [
             "files": [f"{examples_path}/example_images/paper_with_text.png"],
         }
     ],
-    [
-        {
-            "text": "The respective main characters of these two movies meet in real life. Imagine their discussion. It should be sassy, and the beginning of a mysterious adventure.",
-            "files": [f"{examples_path}/example_images/barbie.jpeg", f"{examples_path}/example_images/oppenheimer.jpeg"],
-        }
-    ],
-    [
-        {
-            "text": "Can you explain this meme?",
-            "files": [f"{examples_path}/example_images/running_girl_meme.webp"],
-        }
-    ],
-    [
-        {
-            "text": "What happens to fish if pelicans increase?",
-            "files": [f"{examples_path}/example_images/ai2d_example_2.jpeg"],
-        }
-    ],
-    [
-        {
-            "text": "Give an art-critic description of this well known painting",
-            "files": [f"{examples_path}/example_images/Van-Gogh-Starry-Night.jpg"],
-        }
-    ],
-    [
-        {
-            "text": "Chase wants to buy 4 kilograms of oval beads and 5 kilograms of star-shaped beads. How much will he spend?",
-            "files": [f"{examples_path}/example_images/mmmu_example.jpeg"],
-        }
-    ],
-    [
-        {
-            "text": "Write an online ad for that product.",
-            "files": [f"{examples_path}/example_images/shampoo.jpg"],
-        }
-    ],
-    [
-        {
-            "text": "Describe this image in detail and explain why it is disturbing.",
-            "files": [f"{examples_path}/example_images/cat_cloud.jpeg"],
-        }
-    ],
-    [
-        {
-            "text": "Why is this image cute?",
-            "files": [
-                f"{examples_path}/example_images/kittens-cats-pet-cute-preview.jpg"
-            ],
-        }
-    ],
-    [
-        {
-            "text": "What is formed by the deposition of either the weathered remains of other rocks?",
-            "files": [f"{examples_path}/example_images/ai2d_example.jpeg"],
-        }
-    ],    
-    [
-        {
-            "text": "What's funny about this image?",
-            "files": [f"{examples_path}/example_images/pope_doudoune.webp"],
-        }
-    ],
-    [
-        {
-            "text": "Can this happen in real life?",
-            "files": [f"{examples_path}/example_images/elephant_spider_web.webp"],
-        }
-    ],
-    [
-        {
-            "text": "What's unusual about this image?",
-            "files": [f"{examples_path}/example_images/dragons_playing.png"],
-        }
-    ],
-    [
-        {
-            "text": "Why is that image comical?",
-            "files": [f"{examples_path}/example_images/eye_glasses.jpeg"],
-        }
-    ],
 ]
 
 BOT_AVATAR = "IDEFICS_logo.png"
+# BOT_AVATAR = "YALLA_logo.png"
 
 
 # Chatbot utils
@@ -265,6 +189,7 @@ def extract_images_from_msg_list(msg_list):
     return all_images
 
 
+# comment this call of spaces.GPU later
 @spaces.GPU(duration=180)
 def model_inference(
     user_prompt,
@@ -419,15 +344,15 @@ chatbot = gr.Chatbot(
     height=450,
 )
 
-with gr.Blocks(
-    fill_height=True,
-    css=""".gradio-container .avatar-container {height: 40px width: 40px !important;} #duplicate-button {margin: auto; color: white; background: #f1a139; border-radius: 100vh; margin-top: 2px; margin-bottom: 2px;}""",
-) as demo:
-
-    gr.Markdown("# 🐶 Hugging Face Idefics2 8B Chatty")
-    gr.Markdown("In this demo you'll be able to chat with [Idefics2-8B-chatty](https://huggingface.co/HuggingFaceM4/idefics2-8b-chatty), a variant of [Idefics2-8B](https://huggingface.co/HuggingFaceM4/idefics2-8b-chatty) further fine-tuned on chat datasets.")
-    gr.Markdown("If you want to learn more about Idefics2 and its variants, you can check our [blog post](https://huggingface.co/blog/idefics2).")
-    gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
+# with gr.Blocks(
+#     fill_height=True, # Use this below !?
+#     css=""".gradio-container .avatar-container {height: 40px width: 40px !important;} #duplicate-button {margin: auto; color: white; background: #f1a139; border-radius: 100vh; margin-top: 2px; margin-bottom: 2px;}""",
+# ) as demo:
+with gr.Blocks() as demo:
+    gr.Markdown("# 🐶 YALLA ")
+    gr.Markdown("In this demo you'll be able to chat with YALLA, a variant of [Idefics2-8B](https://huggingface.co/HuggingFaceM4/idefics2-8b-chatty) further fine-tuned on chat datasets, and Moroccan culture 🇲🇦")
+    # gr.Markdown("If you want to learn more about Idefics2 and its variants, you can check our [blog post](https://huggingface.co/blog/idefics2).")
+    # gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
     # model selector should be set to `visbile=False` ultimately
     with gr.Row(elem_id="model_selector_row"):
         model_selector = gr.Dropdown(