Update app.py

app.py

@@ -44,7 +44,7 @@ if uploaded_img is not None:
     layer_ind = options.index(selected_option)
     selected_layer = conv_layers[layer_ind]
 
-    func_model = Model(inputs = model.input, outputs = selected_layer.output)
+    func_model = Model(inputs = model.layers[0].input, outputs = selected_layer.output)
 
     fm = func_model.predict(input_img)
     fm = fm[0]