# app.py
import tensorflow as tf
import numpy as np
from PIL import Image

model = tf.keras.models.load_model('.')

def predict(image):
    # Preprocess image
    image = image.resize((224, 224))
    image_array = np.array(image) / 255.0
    image_array = np.expand_dims(image_array, axis=0)
    
    # Predict
    predictions = model.predict(image_array)
    return predictions[0]