Update app.py

Changed way of loading model

app.py

@@ -48,8 +48,24 @@ uploaded_file = st.file_uploader("Please upload an image", type=["jpeg", "jpg",
 
 if uploaded_file is not None:
     #load models
-    model_M1 = tf.keras.layers.TFSMLayer("IMVIP_Supplementary_Material/models/model1/") #tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/")
-    model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/")
+    model_path1 = "IMVIP_Supplementary_Material/models/model1/"
+    model_path2 = "IMVIP_Supplementary_Material/models/model2/"
+    
+    tfsm_layer1 = tf.keras.layers.TFSMLayer(model_path1, call_endpoint='serving_default')
+    tfsm_layer2 = tf.keras.layers.TFSMLayer(model_path2, call_endpoint='serving_default')
+    
+    input_shape = (256, 256, 3)
+    inputs = Input(shape=input_shape)
+
+    #create the model
+    outputs1 = tfsm_layer1(inputs)
+    model_M1 = Model(inputs, outputs1)
+
+    outputs2 = tfsm_layer2(inputs)
+    model_M2 = Model(inputs, outputs2)
+    
+    #model_M1 = tf.keras.layers.TFSMLayer("IMVIP_Supplementary_Material/models/model1/") #tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/")
+    #model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/")
 
     # Convert the file to an opencv image.
     file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)