tea.yaml

# https://tea.xyz/what-is-this-file
---
version: 1.0.0
codeOwners:
  - '0x26cE9841FfC59dBF1054d1A0799F5eAe02664ef0'
  - '0xcC9E291C5f3D14Fa5056B55FDBcDAeD2F8324042'
quorum: 1
from lobe import ImageModel

model = ImageModel.load('path/to/exported/model/folder')

# OPTION 1: Predict from an image file
result = model.predict_from_file('path/to/file.jpg')

# OPTION 2: Predict from an image url
result = model.predict_from_url('http://url/to/file.jpg')

# OPTION 3: Predict from Pillow image
from PIL import Image
img = Image.open('path/to/file.jpg')
result = model.predict(img)

# Print top prediction
print(result.prediction)

# Print all classes
for label, confidence in result.labels:
    print(f"{label}: {confidence*100}%")

# Visualize the heatmap of the prediction on the image 
# this shows where the model was looking to make its prediction.
heatmap = model.visualize(img)
heatmap.show()