Update app.py

app.py

@@ -1,52 +1,55 @@
-import os
-import gradio as gr
-import requests
-import base64
-from io import BytesIO
-from PIL import Image
-
-def image_to_base64(image):
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    return base64.b64encode(buffered.getvalue()).decode('utf-8')
-
-def base64_to_image(base64_str):
-    return Image.open(BytesIO(base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))))
-
-def search_face(file):
-    url = os.environ.get("SERVER_URL")
-
-    try:
-        image = Image.open(file)
-        image_base64 = image_to_base64(image)
-        r = requests.post(url=url, json={"token": os.environ.get("ACCESS_TOKEN"), "type": "novip", "image": image_base64})
-    except:
-        raise gr.Error("Please select image file!")
-    
-    if r.text == "No matches":
-        gr.Info("No images found.")
-        return []
-    
-    try:
-        res = r.json().get('img_array')
-        out_array = []
-        for item in res:
-            out_array.append((base64_to_image(item["image"]), item["url"] + "*********"))
-        return out_array
-    except:
-        raise gr.Error("Try again.")
-
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column(scale=1):
-            image = gr.Image(type='filepath', height=480)
-            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], 
-                        inputs=image)
-            search_face_button = gr.Button("Search Face")
-        with gr.Column(scale=2):
-            output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
-
-    search_face_button.click(search_face, inputs=[image], outputs=[output])
-
+import os
+import gradio as gr
+import requests
+import base64
+from io import BytesIO
+from PIL import Image
+
+def image_to_base64(image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+def base64_to_image(base64_str):
+    return Image.open(BytesIO(base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))))
+
+def search_face(file):
+    url = os.environ.get("SERVER_URL")
+
+    try:
+        image = Image.open(file)
+        image_base64 = image_to_base64(image)
+        r = requests.post(url=url, json={"token": os.environ.get("ACCESS_TOKEN"), "type": "novip", "image": image_base64})
+    except:
+        raise gr.Error("Please select image file!")
+    
+    if r.text == "No matches":
+        gr.Info("No images found.")
+        return []
+    
+    try:
+        res = r.json().get('img_array')
+        out_array = []
+        for item in res:
+            out_array.append((base64_to_image(item["image"]), item["url"] + "*********"))
+        return out_array
+    except:
+        raise gr.Error("Try again.")
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1):
+            image = gr.Image(type='filepath', height=480)
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], 
+                        inputs=image)
+            search_face_button = gr.Button("Search Face")
+        with gr.Column(scale=2):
+            output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
+
+    search_face_button.click(search_face, inputs=[image], outputs=[output])
+
+    gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')
+
+
 demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)