Update app_dialogue.py

app_dialogue.py

@@ -554,25 +554,25 @@ def model_generation(
 
     return generated_text
 
-def process_example(message, image):
-    print("********* process_example **********")
-    clear_msg, image_value, chat = model_inference(
-        user_prompt_str=message,
-        chat_history=[],
-        image=image,
-        decoding_strategy="Greedy",
-        num_beams=None,
-        temperature=None,
-        no_repeat_ngram_size=None,
-        max_new_tokens=512,
-        min_length=16,
-        repetition_penalty=None,
-        length_penalty=None,
-        top_k=None,
-        top_p=0.95,
-        penalty_alpha=None,
-    )
-    return clear_msg, image_value, chat
+    def process_example(message, image):
+        print("********* process_example **********")
+        clear_msg, image_value, chat = model_inference(
+            user_prompt_str=message,
+            chat_history=[],
+            image=image,
+            decoding_strategy="Greedy",
+            num_beams=None,
+            temperature=None,
+            no_repeat_ngram_size=None,
+            max_new_tokens=512,
+            min_length=16,
+            repetition_penalty=None,
+            length_penalty=None,
+            top_k=None,
+            top_p=0.95,
+            penalty_alpha=None,
+        )
+        return clear_msg, image_value, chat
         
 dope_callback = gr.CSVLogger()
 dope_hf_callback = gr.HuggingFaceDatasetSaver(
@@ -1045,52 +1045,52 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
         examples=[
             ["What are the armed baguettes guarding?", f"{examples_path}/example_images/baguettes_guarding_paris.png"],
             [
-                 "Can you tell me a very short story based on this image?",
-                 f"{examples_path}/example_images/chicken_on_money.png",
-             ],
-            # ["Can you describe the image?", f"{examples_path}/example_images/bear_costume.png"],
-            # ["What is this animal and why is it unusual?", f"{examples_path}/example_images/blue_dog.png"],
-            # [
-            #     "What is this object and do you think it is horrifying?",
-            #     f"{examples_path}/example_images/can_horror.png",
-            # ],
-            # ["What is this sketch for? How would you make an argument to prove this sketch was made by Picasso himself?", f"{examples_path}/example_images/cat_sketch.png"],
-            # ["Which celebrity does this claymation figure look like?", f"{examples_path}/example_images/kanye.jpg"],
-            # [
-            #     "Which famous person does the person in the image look like? Could you craft an engaging narrative featuring this character from the image as the main protagonist?",
-            #     f"{examples_path}/example_images/obama-harry-potter.jpg",
-            # ],
-            # [
-            #     "Is there a celebrity look-alike in this image? What is happening to the person?",
-            #     f"{examples_path}/example_images/ryan-reynolds-borg.jpg",
-            # ],
-            # ["Can you describe this image in details please?", f"{examples_path}/example_images/dragons_playing.png"],
-            # ["What can you tell me about the cap in this image?", f"{examples_path}/example_images/ironman_cap.png"],
-            # [
-            #     "Can you write an advertisement for Coca-Cola based on this image?",
-            #     f"{examples_path}/example_images/polar_bear_coke.png",
-            # ],
-            # [
-            #     "What is the rabbit doing in this image? Do you think this image is real?",
-            #     f"{examples_path}/example_images/rabbit_force.png",
-            # ],
-            # ["What is happening in this image and why is it unusual?", f"{examples_path}/example_images/ramen.png"],
-            # [
-            #     "What I should look most forward to when I visit this place?",
-            #     f"{examples_path}/example_images/tree_fortress.jpg",
-            # ],
-            # ["Who is the person in the image and what is he doing?", f"{examples_path}/example_images/tom-cruise-astronaut-pegasus.jpg"],
-            # [
-            #     "What is happening in this image? Which famous personality does this person in center looks like?",
-            #     f"{examples_path}/example_images/gandhi_selfie.jpg",
-            # ],
-            # [
-            #     (
-            #         "<fake_token_around_image><image:https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/stable-diffusion-xl-coreml/a_high_quality_photo_of_a_surfing_dog.7667.final_float16_original.jpg><fake_token_around_image>What"
-            #         " do you think the dog is doing and is it unusual?"
-            #     ),
-            #     None,
-            # ],
+                "Can you tell me a very short story based on this image?",
+                f"{examples_path}/example_images/chicken_on_money.png",
+            ],
+            ["Can you describe the image?", f"{examples_path}/example_images/bear_costume.png"],
+            ["What is this animal and why is it unusual?", f"{examples_path}/example_images/blue_dog.png"],
+            [
+                "What is this object and do you think it is horrifying?",
+                f"{examples_path}/example_images/can_horror.png",
+            ],
+            ["What is this sketch for? How would you make an argument to prove this sketch was made by Picasso himself?", f"{examples_path}/example_images/cat_sketch.png"],
+            ["Which celebrity does this claymation figure look like?", f"{examples_path}/example_images/kanye.jpg"],
+            [
+                "Which famous person does the person in the image look like? Could you craft an engaging narrative featuring this character from the image as the main protagonist?",
+                f"{examples_path}/example_images/obama-harry-potter.jpg",
+            ],
+            [
+                "Is there a celebrity look-alike in this image? What is happening to the person?",
+                f"{examples_path}/example_images/ryan-reynolds-borg.jpg",
+            ],
+            ["Can you describe this image in details please?", f"{examples_path}/example_images/dragons_playing.png"],
+            ["What can you tell me about the cap in this image?", f"{examples_path}/example_images/ironman_cap.png"],
+            [
+                "Can you write an advertisement for Coca-Cola based on this image?",
+                f"{examples_path}/example_images/polar_bear_coke.png",
+            ],
+            [
+                "What is the rabbit doing in this image? Do you think this image is real?",
+                f"{examples_path}/example_images/rabbit_force.png",
+            ],
+            ["What is happening in this image and why is it unusual?", f"{examples_path}/example_images/ramen.png"],
+            [
+                "What I should look most forward to when I visit this place?",
+                f"{examples_path}/example_images/tree_fortress.jpg",
+            ],
+            ["Who is the person in the image and what is he doing?", f"{examples_path}/example_images/tom-cruise-astronaut-pegasus.jpg"],
+            [
+                "What is happening in this image? Which famous personality does this person in center looks like?",
+                f"{examples_path}/example_images/gandhi_selfie.jpg",
+            ],
+            [
+                (
+                    "<fake_token_around_image><image:https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/stable-diffusion-xl-coreml/a_high_quality_photo_of_a_surfing_dog.7667.final_float16_original.jpg><fake_token_around_image>What"
+                    " do you think the dog is doing and is it unusual?"
+                ),
+                None,
+            ],
         ],
         inputs=[textbox, imagebox],
         outputs=[textbox, imagebox, chatbot],