Spaces:
Sleeping
Sleeping
File size: 921 Bytes
a1e563e 4a5a2fa a1e563e 4a5a2fa 1485906 a1e563e 1485906 a1e563e 1485906 a4404bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# ToDo
- [x] 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`)
- [x] Populate `requirements.txt` with the collected versions.
- [x] Implement `app.py` to run the **YUGOROU/Soba-or-Udon-vit** model from Hugging Face using the `transformers` library.
- Model repo: <https://huggingface.co/YUGOROU/Soba-or-Udon-vit>
- Load weights from `model.safetensors` and base model `google/vit-base-patch16-224`.
- Create an image classification pipeline with PyTorch backend.
- [x] 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.
- [x] Ensure the app can run smoothly in a Hugging Face Spaces environment.
|