Upload folder using huggingface_hub
Browse files- .gitattributes +1 -1
- README.md +7 -56
- best_model.keras +3 -0
- model_config.pkl +3 -0
- tokenizer.pkl +3 -0
.gitattributes
CHANGED
|
@@ -33,4 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
-
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
best_model.keras filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -1,57 +1,8 @@
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
pinned: false
|
| 10 |
-
---
|
| 11 |
-
|
| 12 |
-
# Image Captioning — Hugging Face Space
|
| 13 |
-
|
| 14 |
-
Triển khai inference cho mô hình image captioning bằng TensorFlow/Keras, với giao diện Gradio đơn giản cho upload ảnh và nhận caption.
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
# Image Captioning — Hugging Face Space
|
| 18 |
-
|
| 19 |
-
Triển khai inference cho mô hình image captioning dùng TensorFlow/Keras, EfficientNetV2B0 và giao diện Gradio.
|
| 20 |
-
|
| 21 |
-
## Cấu trúc tệp cần có
|
| 22 |
-
```text
|
| 23 |
-
.
|
| 24 |
-
├── app.py # UI Gradio cho Hugging Face Space
|
| 25 |
-
├── flickr30k.py # Logic model + tiền xử lý (đã cung cấp)
|
| 26 |
-
├── best_model.keras # Trọng số mô hình (đặt cùng thư mục)
|
| 27 |
-
├── tokenizer.pkl # Tokenizer đã fit
|
| 28 |
-
├── model_config.pkl # Chứa max_length, vocab_size
|
| 29 |
-
├── requirements.txt
|
| 30 |
-
└── README.md
|
| 31 |
-
```
|
| 32 |
-
|
| 33 |
-
Các hàm sử dụng trực tiếp từ `flickr30k.py`: `load_caption_model`, `load_tokenizer_and_config`, `load_feature_extractor`, `extract_features_from_image`, `generate_caption`.
|
| 34 |
-
|
| 35 |
-
## Chạy cục bộ
|
| 36 |
-
```bash
|
| 37 |
-
python -m venv .venv
|
| 38 |
-
. .venv/bin/activate # Windows: .venv\Scripts\activate
|
| 39 |
-
pip install --upgrade pip
|
| 40 |
-
pip install -r requirements.txt
|
| 41 |
-
|
| 42 |
-
# Đảm bảo 3 tệp đã có:
|
| 43 |
-
# best_model.keras, tokenizer.pkl, model_config.pkl
|
| 44 |
-
|
| 45 |
-
python app.py
|
| 46 |
-
```
|
| 47 |
-
Mở URL Gradio hiển thị trong terminal.
|
| 48 |
-
|
| 49 |
-
## Triển khai lên Hugging Face Spaces
|
| 50 |
-
1) Tạo Space mới: SDK = Gradio, chọn CPU hoặc GPU tùy trọng số.
|
| 51 |
-
2) Đẩy các tệp: `app.py`, `flickr30k.py`, `requirements.txt`, `README.md`, và 3 tệp trọng số/cấu hình.
|
| 52 |
-
3) Sau khi build hoàn tất, Space sẽ mở UI upload ảnh và trả caption.
|
| 53 |
-
|
| 54 |
-
## Ghi chú tương thích
|
| 55 |
-
- Mặc định dùng `tensorflow==2.12.0`. Nếu bạn dùng trọng số huấn luyện ở phiên bản khác, cần đồng bộ phiên bản TensorFlow/Keras tương ứng.
|
| 56 |
-
- Sử dụng `opencv-python-headless` thay vì `opencv-python` để tránh lỗi GUI trên môi trường server.
|
| 57 |
-
- Nếu thiếu tài nguyên trên Space Free, hạ kích thước mô hình hoặc chuyển phần cứng sang GPU trả phí.
|
|
|
|
| 1 |
---
|
| 2 |
+
datasets:
|
| 3 |
+
- nlphuji/flickr30k
|
| 4 |
+
language:
|
| 5 |
+
- en
|
| 6 |
+
metrics:
|
| 7 |
+
- bleu
|
| 8 |
+
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
best_model.keras
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:126f6eaffa59549b492ed3d92eb7a1cb60aea13d925a4f465982f7c13e8f1047
|
| 3 |
+
size 64213314
|
model_config.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c488749928c2e33ea55133d4a9e4d582c29377dd7322d28fb9b916be257bcde1
|
| 3 |
+
size 47
|
tokenizer.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3f53ea4f0f9ba9f4a366454336b8e7ec87c3537160477075decc50db052c4733
|
| 3 |
+
size 716347
|