fuyu-compare / app.py
tlodato
adding compare code
4ebcacd
raw
history blame
1.38 kB
#app.py
import os
import gradio as gr
from gradio_client import Client
from gradio_client.client import Job
def vqa_fuyu(image_path: str, question: str, return_job: bool = False) -> str | Job:
try:
client = Client("https://adept-fuyu-8b-demo.hf.space/--replicas/9kcqv/")
fn = client.submit if return_job else client.predict
return fn(image_path, question, fn_index=3)
except Exception:
gr.Warning("The Fuyu-8B Space is currently unavailable. Please try again later.")
return ""
def answer_question(image_path: str, question:str) -> tuple[str]:
jobs = [
vqa_fuyu(image_path, question, return_job=True),
]
return tuple(job.result() if job else "" for job in jobs)
def ask(q):
return q
with gr.Blocks(css="style.css") as demo:
gr.Markdown(DESCRIPTION)
with gr.Row():
with gr.Column():
input_image = gr.Image(type="filepath")
text_input = gr.Interface(fn=ask, inputs="text", outputs="text")
run_button = gr.Button("ask question")
with gr.Column():
out_fuyu = gr.Textbox(label="Fuyu-8B")
outputs = [
out_fuyu,
]
run_button.click(
fn=answer_question,
inputs=[input_image,text_input],
outputs=outputs,
api_name="vqa",
)
if __name__ == "__main__":
demo.queue(max_size=20).launch()