inference.py

# Performs inference of hot dog or not hot dog
from transformers import pipeline

pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")

def predict_hotdog(image):
  predictions = pipeline(image)
  return {p["label"]: p["score"] for p in predictions}