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.