import gradio as gr
from transformers import pipeline

import os

env_vars = os.environ

# 打印所有环境变量
for key, value in env_vars.items():
    print(f"{key}: {value}")


pipe = pipeline("object-detection", model="Charles95/autotrain-detr-cppe-v5")

examples = [
    ["images/1.jpg"],
    ["images/2.jpg"],
    ["images/3.jpg"],
]

demo = gr.Interface.from_pipeline(pipe, examples=examples)

demo.launch(server_name="0.0.0.0")