File size: 770 Bytes
cc9720d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import importlib.metadata
import gradio as gr
from optimum_support import show_is_supported
with gr.Blocks() as app:
gr.Markdown("# Check if model is supported by optimum-intel[openvino]")
with gr.Column():
model_id = gr.Textbox(label="model_id")
process_button = gr.Button("Check")
output_text = gr.Markdown(label="result", height=100)
optimum_intel_version = importlib.metadata.version("optimum-intel")
gr.Markdown(
f"Tested with optimum-intel {optimum_intel_version}. For testing purposes only, results may be wrong."
)
process_button.click(show_is_supported, inputs=[model_id], outputs=output_text)
model_id.submit(show_is_supported, inputs=[model_id], outputs=output_text)
app.launch()
|