Update src/app.py

src/app.py

@@ -1,4 +1,6 @@
 # Code developed by: Ruslan Magana Vsevolodovna
+"""Template Demo for IBM Granite Hugging Face spaces."""
+
 from collections.abc import Iterator
 from datetime import datetime
 from pathlib import Path
@@ -59,8 +61,13 @@ text_tokenizer.use_default_system_prompt = False
 vision_model_path = "ibm-granite/granite-vision-3.1-2b-preview"
 vision_processor = LlavaNextProcessor.from_pretrained(vision_model_path, use_fast=True)
 vision_model = LlavaNextForConditionalGeneration.from_pretrained(
-    vision_model_path, torch_dtype=torch.float16, device_map="auto", low_cpu_mem_usage=True
-) # Added low_cpu_mem_usage and explicit torch_dtype
+    vision_model_path,
+    torch_dtype=torch.float16,
+    device_map="auto",
+    low_cpu_mem_usage=True,
+    force_download=True, # Added force_download to ensure fresh download
+    revision="main" # Added revision to specify main branch (if needed)
+)
 
 
 @spaces.GPU
@@ -316,6 +323,13 @@ with gr.Blocks(fill_height=True, css_paths=css_file_path, head_paths=head_file_p
     )
 
 
+    image_examples_dir = Path(__file__).parent / "image_examples" # Create a folder 'image_examples' in the same directory as app.py and put images there
+    image_examples = [
+        str(image_examples_dir / "cat.jpg"), # Replace "cat.jpg" with actual image file names in 'image_examples'
+        str(image_examples_dir / "dog.jpg"), # Replace "dog.jpg" with actual image file names in 'image_examples'
+        str(image_examples_dir / "horse.jpg") # Replace "horse.jpg" with actual image file names in 'image_examples'
+    ]
+
     gr.Examples(
         examples=[
             ["Explain the concept of quantum computing to someone with no background in physics or computer science.", None],
@@ -375,7 +389,9 @@ The Pair class is a simple data structure that stores two integers.
 
 The main method creates an array of integers, initializes the difference value, and calls the findPairs method to find all pairs in the array. Finally, the code iterates over the list of pairs and prints each pair to the console.""" , None # noqa: E501
             ],
-            ["What is in this image?", "https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png"] # Vision example
+            ["What is in this image?", image_examples[0]], # Vision example using local image
+            ["Describe this image in detail", image_examples[1]], # Vision example using local image
+            ["Identify the object in the image", image_examples[2]], # Vision example using local image
         ],
         inputs=[text_input, image_input],
         example_labels=[
@@ -385,7 +401,9 @@ The main method creates an array of integers, initializes the difference value,
             "Boosting productivity habits",
             "Explain and document your code",
             "Generate Java Code",
-            "Vision Example: What is in this image?"
+            "Vision Example 1: What is in this image?",
+            "Vision Example 2: Describe this image",
+            "Vision Example 3: Identify object",
         ],
         cache_examples=False,
     )