Update inference.py

inference.py

@@ -1,3 +1,17 @@
 import streamlit as st
 def demo():
-    st.write("yes its work")
+    st.write("yes its work")
+def run_example(image, model, processor, task_prompt, text_input=None):
+    
+    inputs = processor(text=text_input, images=image, return_tensors="pt")
+    generated_ids = model.generate(
+        input_ids=inputs["input_ids"],
+        pixel_values=inputs["pixel_values"],
+        max_new_tokens=1024,
+        num_beams=4
+    )
+    generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    print("generated_text:",generated_text)
+    parsed_answer = processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.width, image.height))
+    return parsed_answer
+