Spaces:
Sleeping
Sleeping
Commit
·
72c24b3
1
Parent(s):
13147dd
Update app.py
Browse files
app.py
CHANGED
@@ -47,21 +47,6 @@ def load_encoder_model():
|
|
47 |
encoder=tf.keras.models.load_model('./encoder_model.h5')
|
48 |
return encoder
|
49 |
|
50 |
-
@st.cache_resource()
|
51 |
-
def feature_extractor():
|
52 |
-
FEATURE_EXTRACTOR=tf.keras.models.load_model('./feature_extractor.h5')
|
53 |
-
return FEATURE_EXTRACTOR
|
54 |
-
|
55 |
-
|
56 |
-
FEATURE_EXTRACTOR = feature_extractor()
|
57 |
-
FEATURE_EXTRACTOR.trainable = False
|
58 |
-
image_input = Input(shape=(IMG_HEIGHT, IMG_WIDTH, IMG_CHANNELS))
|
59 |
-
image_features = FEATURE_EXTRACTOR(image_input)
|
60 |
-
x = Reshape((FEATURES_SHAPE[0] * FEATURES_SHAPE[1], FEATURES_SHAPE[2]))(image_features)
|
61 |
-
encoder_output = Dense(ATTENTION_DIM, activation="relu")(x)
|
62 |
-
encoder = tf.keras.Model(inputs=image_input, outputs=encoder_output)
|
63 |
-
|
64 |
-
|
65 |
|
66 |
st.title(":blue[Nishant Guvvada's] :red[AI Journey] Image Caption Generation")
|
67 |
image = Image.open('./title.jpg')
|
|
|
47 |
encoder=tf.keras.models.load_model('./encoder_model.h5')
|
48 |
return encoder
|
49 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
50 |
|
51 |
st.title(":blue[Nishant Guvvada's] :red[AI Journey] Image Caption Generation")
|
52 |
image = Image.open('./title.jpg')
|