Spaces:
Sleeping
Sleeping
A newer version of the Gradio SDK is available:
5.46.1
ToDo
- Determine the latest stable versions of the following libraries:
transformers
(use latest)gradio
(example: 5.36.2 is currently latest)torch
(avoid 2.3.x)- other helpful packages (e.g.
Pillow
,safetensors
)
- Populate
requirements.txt
with the collected versions. - Implement
app.py
to run the YUGOROU/Soba-or-Udon-vit model from Hugging Face using thetransformers
library.- Model repo: https://huggingface.co/YUGOROU/Soba-or-Udon-vit
- Load weights from
model.safetensors
and base modelgoogle/vit-base-patch16-224
. - Create an image classification pipeline with PyTorch backend.
- Build a simple Gradio interface for image classification.
- Add a title and a short description at the top of the UI.
- Accept an image input and show the predicted label as output.
- Ensure the app can run smoothly in a Hugging Face Spaces environment.