lordpotato
commited on
Commit
·
2ed5ed5
1
Parent(s):
9f5df1c
revised project strucutre, added requirements, need to add 2
Browse files- README.md +4 -0
- app.py +1 -1
- caption_model.h5 → models/caption_model.h5 +0 -0
- image-caption.ipynb → notebooks/image-caption.ipynb +0 -0
- requirements.txt +5 -0
README.md
CHANGED
|
@@ -10,3 +10,7 @@ pinned: false
|
|
| 10 |
---
|
| 11 |
|
| 12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
Model trained on flickr8k : ResNet + LSTM
|
| 16 |
+
on coco2017 (using 8k images) : CLIP + GPT2
|
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from tensorflow.keras.applications.inception_v3 import preprocess_input
|
|
| 9 |
import re
|
| 10 |
|
| 11 |
# Load the model
|
| 12 |
-
model = tf.keras.models.load_model('caption_model.h5')
|
| 13 |
|
| 14 |
# Load tokenizer (you'll need to adapt this to your actual tokenizer loading)
|
| 15 |
# Replace with your actual tokenizer loading
|
|
|
|
| 9 |
import re
|
| 10 |
|
| 11 |
# Load the model
|
| 12 |
+
model = tf.keras.models.load_model('models/caption_model.h5')
|
| 13 |
|
| 14 |
# Load tokenizer (you'll need to adapt this to your actual tokenizer loading)
|
| 15 |
# Replace with your actual tokenizer loading
|
caption_model.h5 → models/caption_model.h5
RENAMED
|
File without changes
|
image-caption.ipynb → notebooks/image-caption.ipynb
RENAMED
|
File without changes
|
requirements.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
gradio
|
| 2 |
+
tensorflow
|
| 3 |
+
numpy
|
| 4 |
+
requests
|
| 5 |
+
pillow
|