NimaBoscarino's picture
WIP: initial version of checks, creating a Gradio UI for the app
25bf2cc
raw history blame
No virus
1.18 kB
import gradio as gr
from compliance_checks import (
ComplianceSuite,
ModelProviderIdentityCheck,
IntendedPurposeCheck,
GeneralLimitationsCheck,
ComputationalRequirementsCheck,
)
from bloom_card import bloom_card
def run_compliance_check(model_card: str):
suite = ComplianceSuite(checks=[
ModelProviderIdentityCheck(),
IntendedPurposeCheck(),
GeneralLimitationsCheck(),
ComputationalRequirementsCheck(),
])
results = suite.run(model_card)
return str([r[0] for r in results])
with gr.Blocks() as demo:
gr.Markdown("""\
# Model Card Validator
Following Article 13 of the EU AI Act
""")
with gr.Row():
with gr.Column():
model_card_box = gr.TextArea()
populate_sample = gr.Button(value="Populate Sample")
submit = gr.Button()
with gr.Column():
results_list = gr.Text()
submit.click(
fn=run_compliance_check,
inputs=[model_card_box],
outputs=[results_list]
)
populate_sample.click(
fn=lambda: bloom_card,
inputs=[],
outputs=[model_card_box]
)
demo.launch()