removed llava onevision

app.py

@@ -291,7 +291,7 @@ with gr.Blocks() as demo:
             activation_map_output = gr.Gallery(label="activation Map", height=300, columns=1)
 
         with gr.Column():
-            model_selector = gr.Dropdown(choices=["Clip", "ChartGemma-3B", "Janus-1B", "Janus-7B", "LLaVA-1.5-7B", "LLaVA-onevision-qwen2-7b-si"], value="Clip", label="model")
+            model_selector = gr.Dropdown(choices=["Clip", "ChartGemma-3B", "Janus-1B", "Janus-7B", "LLaVA-1.5-7B"], value="Clip", label="model")
             response_type = gr.Dropdown(choices=["Visualization only"], value="Visualization only", label="response_type")
             focus = gr.Dropdown(choices=["Visual Encoder"], value="Visual Encoder", label="focus")
             activation_map_method = gr.Dropdown(choices=["GradCAM"], value="GradCAM", label="activation map type")
@@ -347,62 +347,65 @@ with gr.Blocks() as demo:
     examples_inpainting = gr.Examples(
         label="Multimodal Understanding examples",
         examples=[
-            
+
             [
-                "What is the approximate global smartphone market share of Samsung?",
-                "images/PieChart.png"
+                "What was the price of a barrel of oil in February 2020?", 
+                "images/LineChart.png" 
             ],
+
             [
                 "What is the average internet speed in Japan?",
                 "images/BarChart.png"
             ],
+
             [
-                "What was the average price of coffee beans in October 2019?",
-                "images/AreaChart.png"
-            ],
-            [
-                "Which city's metro system has the largest number of stations?", 
-                "images/BubbleChart.png"
+                "What is the cost of peanuts in Seoul?", 
+                "images/StackedBar.png"
             ],
 
             [ 
-                "True/False: In 2020, the unemployment rate for Washington (WA) was higher than that of Wisconsin (WI).", 
-                "images/Choropleth_New.png"
+                "Which country has the lowest proportion of Gold medals?", 
+                "images/Stacked100.png"
             ],
-
+            
+            [
+                "What is the approximate global smartphone market share of Samsung?",
+                "images/PieChart.png"
+            ],
+            
             [ 
                 "What distance have customers traveled in the taxi the most?", 
                 "images/Histogram.png"
             ],
 
             [
-                "What was the price of a barrel of oil in February 2020?", 
-                "images/LineChart.png" 
+                "True/False: There is a negative linear relationship between the height and the weight of the 85 males.", 
+                "images/Scatterplot.png"
             ],
 
             [
-                "True/False: eBay is nested in the Software category.", 
-                "images/TreeMap.png"
+                "What was the average price of pount of coffee beans in October 2019?",
+                "images/AreaChart.png"
             ],
 
             [
-                "True/False: There is a negative linear relationship between the height and the weight of the 85 males.", 
-                "images/Scatterplot.png"
-            ],
-            
-            [ 
-                "Which country has the lowest proportion of Gold medals?", 
-                "images/Stacked100.png"
+                "What was the ratio of girls named 'Isla' to girls named 'Amelia' in 2012 in the UK?", 
+                "images/StackedArea.png"
             ],
 
             [
-                "What was the ratio of girls named 'Isla' to girls named 'Amelia' in 2012 in the UK?", 
-                "images/StackedArea.png"
+                "Which city's metro system has the largest number of stations?", 
+                "images/BubbleChart.png"
+            ],
+
+            [ 
+                "True/False: In 2020, the unemployment rate for Washington (WA) was higher than that of Wisconsin (WI).", 
+                "images/Choropleth_New.png"
             ],
 
             [
-                "What is the cost of peanuts in Seoul?", 
-                "images/StackedBar.png"
+                "True/False: eBay is nested in the Software category.", 
+                "images/TreeMap.png"
             ]
             
         ],

demo/model_utils.py

@@ -162,8 +162,9 @@ class LLaVA_Utils(Model_Utils):
                 ],
             },
         ]
+        
         prompt = self.processor.apply_chat_template(conversation, add_generation_prompt=True)
-        pil_images = [Image.fromarray(image)]
+        pil_images = [Image.fromarray(image).resize((384, 384))]
         prepare_inputs = self.processor(
             images=pil_images, text=prompt, return_tensors="pt"
         ).to(self.cuda_device, dtype=self.dtype)