Update app.py

app.py

@@ -10,12 +10,14 @@ import os
 from PIL import Image
 import numpy as np
 import gradio as gr
+import tensorflow as tf
 
 model = Xception()
 
 # Restructure model
 model = Model(inputs = model.inputs , outputs = model.layers[-2].output)
 
+caption_model = tf.keras.models.load_model('model.h5')
 
 with open('captions.txt', 'r') as f:
     next(f)