phuoc60648 commited on
Commit
d6f45bc
·
verified ·
1 Parent(s): b95b172

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. .gitattributes +1 -1
  2. README.md +7 -56
  3. best_model.keras +3 -0
  4. model_config.pkl +3 -0
  5. 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
- img_cap/best_model.keras 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
+ best_model.keras filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,57 +1,8 @@
1
  ---
2
- title: Image Captioning
3
- emoji: 🖼️
4
- colorFrom: blue
5
- colorTo: pink
6
- sdk: gradio
7
- sdk_version: "4.40.0"
8
- app_file: app.py
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