Update app.py

app.py

@@ -49,7 +49,6 @@ def load_encoder_model():
 
     
 st.title(":blue[Nishant Guvvada's] :red[AI Journey]  Image Caption Generation")
-image = Image.open('./title.jpg')
 st.image(image)
 st.write("""
          # Multi-Modal Machine Learning
@@ -78,6 +77,11 @@ word_to_index = StringLookup(
     mask_token="", vocabulary=tokenizer
 )
 
+# Lookup table: Index -> Word
+index_to_word = StringLookup(
+    mask_token="", vocabulary=tokenizer.get_vocabulary(), invert=True
+)
+
 
 ## Probabilistic prediction using the trained model
 def predict_caption(file):