Update app.py

app.py

@@ -13,7 +13,7 @@ user_embeddings = {}
 
 # Preprocess the image
 def preprocess_image(image):
-    image = cv2.resize(image, (200, 200))  # Assuming your model expects 200x200 input
+    image = cv2.resize(image, (200, 200))  # Resize image to 200x200
     image = tf.keras.applications.resnet50.preprocess_input(image)
     return np.expand_dims(image, axis=0)
 
@@ -45,7 +45,7 @@ def recognize_user(image):
 # Gradio interface for registering users
 register_interface = gr.Interface(
     fn=register_user,
-    inputs=[gr.Image(shape=(200, 200)), gr.Textbox(label="User ID")],
+    inputs=[gr.Image(), gr.Textbox(label="User ID")],
     outputs="text",
     live=True
 )
@@ -53,7 +53,7 @@ register_interface = gr.Interface(
 # Gradio interface for recognizing users
 recognize_interface = gr.Interface(
     fn=recognize_user,
-    inputs=gr.Image(shape=(200, 200)),
+    inputs=gr.Image(),
     outputs="text",
     live=True
 )