Update app.py

app.py

@@ -31,6 +31,21 @@ Given the question, chart analysis output answer and additional context, come up
 """
 
 
+
+def function_that_does_something(**kwargs):
+    # Extract the first two items from the kwargs dictionary
+    first_two_items = dict(list(kwargs.items())[:2])
+    
+    # Extract the 'three' argument, defaulting to 3 if not provided
+    three = kwargs.get('three', 3)
+    
+    # Call another function with the filtered keyword arguments
+    result = function_of_interest(**first_two_items)
+    
+    # Return both the result of function_of_interest and the 'three' argument
+    return result, three
+
+
 @spaces.GPU
 def predict(image, input_text, input_context):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -44,7 +59,7 @@ def predict(image, input_text, input_context):
     prompt_length = inputs['input_ids'].shape[1]
     
     # Generate chart text
-    generate_ids = model.generate(image=image, input_text=input_text, max_new_tokens=512)
+    generate_ids = model.generate(**inputs, max_new_tokens=512)
     output_text = processor.batch_decode(generate_ids[:, prompt_length:], skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
 
     # Generate next layer prompt (WIP)