changed detector backend to retinaface and changed output message of save identity

app.py

@@ -16,14 +16,14 @@ debug=False
 
 def save_identity(image , name):
   try:
-    embeddings = DeepFace.represent(image , model_name=model_name)
+    embeddings = DeepFace.represent(image , model_name=model_name , detector_backend = "retinaface")
     embeddings = embeddings[0]
 
     identity = {"embeddings":embeddings["embedding"] , "name" : name }
 
     result = identities_collection.insert_one(identity)
 
-    return str(result)
+    return f"{name} stored in database successfully.It is recommended to add 2 or 3 high quality images for one person"
   except Exception as error:
     return str(error)
 
@@ -69,6 +69,7 @@ def predict_image(image):
     target_embedding_array = DeepFace.represent(
         img_path=image,
         model_name=model_name,
+        detector_backend = "retinaface"
     )
     identities = []
     # for each face compare its embeddings with all face embeddings in database
@@ -116,7 +117,7 @@ def predict_image(image):
 
 # Create Gradio interfaces for input and output
 image_input = gr.inputs.Image(shape=(160, 160))
-label_input = gr.inputs.Textbox(label="Enter Label")
+label_input = gr.inputs.Textbox(label="Enter Name")
 label_output = gr.outputs.Textbox()
 
 
@@ -131,7 +132,7 @@ interface1 = gr.Interface(
 
 
 # Create Gradio interfaces for image input and output
-image_input2 = gr.inputs.Image(shape=(None, None))
+image_input2 = gr.inputs.Image(shape=(160,160))
 output_image = gr.outputs.Image(type="numpy")
 # output_image = gr.outputs.Textbox()
 
@@ -155,4 +156,4 @@ gr.TabbedInterface(
 ).queue().launch()
 
 # Launch the Gradio interface
-interface.launch()
+# interface.launch()