FastAPI-powered backend for gemstone image classification using a pretrained ViT model from Hugging Face.
π― This API receives an image of a gemstone and returns the predicted label using computer vision.
The backend uses:
google/vit-base-patch16-224
Vision Transformer (Hugging Face)FastAPI
for serving the APIUvicorn
as the ASGI server
Send a POST request with a JPG/PNG image: