| from fastapi import FastAPI, File, UploadFile |
| import numpy as np |
| from PIL import Image |
| import io |
| import tensorflow as tf |
|
|
| model = tf.keras.models.load_model('946_.keras', compile=False) |
|
|
|
|
| app = FastAPI() |
|
|
| @app.post("/classify") |
| async def classify(image: UploadFile = File(...)): |
| if image is not None: |
| img = Image.open(io.BytesIO(await image.read())) |
| img = img.resize((64,64)) |
| img_array = np.array(img) / 255.0 |
| img_array = np.expand_dims(img_array, axis=0) |
| predictions = model.predict(img_array) |
| predicted_class_idx = np.argmax(predictions) |
| predicted_class_idx = int(predicted_class_idx) |
| return {"prediction": predicted_class_idx} |
| else: |
| return {"error": "No image provided"} |
