Spaces:
Runtime error
Runtime error
| ''''' | |
| @author : Ali Mustofa HALOTEC | |
| @module : Service Models OCR Container Number | |
| @Created on : 29 Nov 2022 | |
| ''' | |
| import os | |
| import io | |
| import json | |
| import uvicorn | |
| from PIL import Image | |
| import numpy as np | |
| from src.process import main_process | |
| from fastapi import FastAPI,File | |
| app = FastAPI() | |
| def home(): | |
| return {"msg": "Avocado ripe classification"} | |
| def fruit_ripe_classification(file: bytes = File(...)): | |
| image = Image.open(io.BytesIO(file)).convert("RGB") | |
| image = np.array(image) | |
| image = image[:,:,::-1].copy() | |
| result = main_process( | |
| image | |
| ) | |
| result_json = result.json() | |
| return json.loads(result_json) | |
| if __name__ == '__main__': | |
| uvicorn.run("api:app", host="0.0.0.0", port=8080, reload=True) | |
| # not_ripe = ["Medium Sea Green", "Viridian", "Medium Jungle Green", "Hooker'S Green", ] | |
| # almost_ripe = ["Aurometalsaurus", "Cadet Blue", "Viridian", "Payne'S Grey"] | |
| # almost/ripe = Hunter Green, Slate Gray | |
| # ripe = ["Ucla Blue", "Cadet Grey", "Dark Slate Gray", "Aurometalsaurus", "Cadet Blue", "Rackley"] | |
| # overripe = ["Dark Jungle Green", "Payne'S Grey", "Aurometalsaurus", "Purple Taupe", "Dim Gray", ] |