food-ai/bentofile.gpu.yaml at main · daviaraujocc/food-ai · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
service: "service.py:FoodGenius"
name: foodgenius-service-gpu
labels:
  owner: daviaraujocc
  project: foodgenius
description: |
    ## 🍽️ FoodGeniusAI: Your AI-Powered Food Classifier 🍽️

    FoodGeniusAI is a powerful AI application built using EfficientNetB2 and PyTorch. It can detect the type of food in an image and determine whether the image contains food or not.

    ### Key Features:
    - **Food or Non-Food Classification**: Using the Food5K dataset.
    - **Food101 Classification**: Identifying 101 different types of food.

    > This version leverages CUDA for faster processing on compatible GPUs.

    > Try it out on [Hugging Face Spaces](https://huggingface.co/spaces/daviaraujocc/foodgeniusai) and explore the code on [GitHub](https://github.com/daviaraujocc/FoodGeniusAI)!
include:
  - "*.py"
  - "*.txt"
  - "models/*.pth"
python:
  requirements_txt: "./requirements/gpu-requirements.txt"
docker:
  cuda_version: 12.1
  dockerfile_template: ./Dockerfile.template
  python_version: 3.11
envs:
  - name: BENTOML_DEVICE
    value: "cuda"
  - name: FOOD101_MODEL_PATH
    value: models/pretrained_effnetb2_food_101.pth
  - name: FOOD_NONFOOD_MODEL_PATH
    value: models/pretrained_effnetb2_food_or_nonfood.pth