food-classifier / app.py
Johnny0619's picture
Update app.py
093c2a6 verified
import gradio as gr
from transformers import pipeline
# モデルの読み込み
classifier = pipeline("image-classification", model="Kaludi/food-category-classification-v2.0")
def predict_food(image):
if image is None:
return None
predictions = classifier(image)
return {p["label"]: p["score"] for p in predictions}
# アプリ設定
demo = gr.Interface(
fn=predict_food,
# 【修正点】 streaming=True を削除しました。
# これにより「動画」としてではなく「パラパラ漫画」のように画像を連続送信する形になり、負荷が下がります。
inputs=gr.Image(sources=["webcam"], label="カメラ"),
outputs=gr.Label(num_top_classes=3, label="リアルタイム判定"),
# live=True にしておくと、画像が変わるたびに判定が走ります
live=True,
title="リアルタイム AI 料理判定",
description="カメラに料理を映してください。"
)
if __name__ == "__main__":
demo.launch()