Update app.py

app.py

@@ -45,8 +45,8 @@ if uploaded_img is not None:
     input_img = img_resized.reshape(1,28,28,1)
     
     #model.predict(np.zeros((1, 28, 28, 1), dtype=np.float32))
-    dummy_input = tf.zeros((1, 28, 28, 1))
-    _ = model(dummy_input)
+    # dummy_input = tf.zeros((1, 28, 28, 1))
+    # _ = model(dummy_input)
 
     st.image(img_resized, caption="Uploaded Image (Resized to 28x28)", use_container_width =True, channels="GRAY")
 
@@ -54,7 +54,7 @@ if uploaded_img is not None:
     selected_layer = conv_layers[layer_ind]
 
     #func_model = Model(inputs = model.layers[0].input, outputs = model.selected_layer.output)
-    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)