Update app.py

app.py

@@ -5,10 +5,16 @@ from PIL import Image
 import urllib.request
 from utils import *
 
-# Initialize and build the model first
-model = model_arc()
-model.load_weights("classify_model.h5")
+def build_and_load_model():
+    # Initialize and build the model first
+    model = model_arc()
+    dummy_input = tf.zeros((1, 256, 256, 3))
+    model.build(input_shape=dummy_input.shape)
+    model(dummy_input)
+    model.load_weights("classify_model.h5")
+    return model
 
+model = build_and_load_model()
 labels = gen_labels()
 
 html_temp = '''