revised project strucutre, added requirements, need to add 2

README.md

@@ -10,3 +10,7 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+Model trained on flickr8k : ResNet + LSTM
+on coco2017 (using 8k images) : CLIP + GPT2

app.py

@@ -9,7 +9,7 @@ from tensorflow.keras.applications.inception_v3 import preprocess_input
 import re
 
 # Load the model
-model = tf.keras.models.load_model('caption_model.h5')
+model = tf.keras.models.load_model('models/caption_model.h5')
 
 # Load tokenizer (you'll need to adapt this to your actual tokenizer loading)
 # Replace with your actual tokenizer loading

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+tensorflow
+numpy
+requests
+pillow