nishantguvvada commited on
Commit
72c24b3
·
1 Parent(s): 13147dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -15
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')