from transformers import pipeline
import gradio as gr

# Hugging Faceの画像分類パイプラインを使用（事前学習モデル）
classifier = pipeline("image-classification", model="google/vit-base-patch16-224")

def classify_image(img):
    results = classifier(img)
    return {r["label"]: float(r["score"]) for r in results}

# Gradioインターフェースの定義
gr.Interface(
    fn=classify_image,
    inputs=gr.Image(type="pil"),
    outputs=gr.Label(num_top_classes=3),
    title="動物種判別ツール",
    description="画像をアップロードすると画像分類モデルで判定します。"
).launch()