import spaces
import gradio as gr
import transformers_gradio

demo = gr.load(
    name='google/paligemma2-10b-ft-docci-448',
    src=transformers_gradio.registry,
)

demo.fn = spaces.GPU()(demo.fn)

demo.launch()