File size: 690 Bytes
7c62949 d01667b 65ecaf3 2164273 65ecaf3 2164273 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import streamlit as st
def demo():
#st.write("yes its work")
pass
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 generated_text
|