Update app.py

app.py

@@ -44,7 +44,7 @@ def predict(image, input_text, input_context):
     prompt_length = inputs['input_ids'].shape[1]
     
     # Generate chart text
-    generate_ids = model.generate(**inputs, max_new_tokens=512)
+    generate_ids = model.generate(image=image, input_text=input_text, 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)