Spaces:
Sleeping
Sleeping
Commit
·
50dbd25
1
Parent(s):
d2ff190
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,6 +30,7 @@ IMG_WIDTH = 299
|
|
| 30 |
IMG_CHANNELS = 3
|
| 31 |
ATTENTION_DIM = 512 # size of dense layer in Attention
|
| 32 |
VOCAB_SIZE = 20000
|
|
|
|
| 33 |
|
| 34 |
@st.cache_resource()
|
| 35 |
def load_image_model():
|
|
|
|
| 30 |
IMG_CHANNELS = 3
|
| 31 |
ATTENTION_DIM = 512 # size of dense layer in Attention
|
| 32 |
VOCAB_SIZE = 20000
|
| 33 |
+
FEATURES_SHAPE = (8, 8, 1536)
|
| 34 |
|
| 35 |
@st.cache_resource()
|
| 36 |
def load_image_model():
|