import gradio as gr
from transformers import pipeline

# 选择 Hugging Face 预训练模型
classifier = pipeline("feature-extraction", model="facebook/deit-base-distilled-patch16-224")

# 定义分类函数
def classify_image(image):
    predictions = classifier(image)
    return {"feature_vector": predictions[0]}  # 返回特征向量

# 创建 Gradio 界面
demo = gr.Interface(fn=classify_image, inputs="image", outputs="text", title="Image Classification Demo")
demo.launch()