Spaces:

Lokesh1024
/

Image_Cap2

Sleeping

Lokesh1024 commited on Jul 27, 2024

Commit

0263963

verified ·

1 Parent(s): fb965d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pickle
 # Load the pre-trained model
 @st.cache(allow_output_mutation=True)
 def load_model():
-    model = tf.keras.models.load_model('caption_model.h5')
     return model
 # Load the tokenizer

 # Load the pre-trained model
 @st.cache(allow_output_mutation=True)
 def load_model():
+    model = tf.keras.models.load_model('./caption_model.h5')
     return model
 # Load the tokenizer