Update app.py

app.py

@@ -18,8 +18,11 @@ STATUS_MESSAGES = {
 }
 
 BACKEND = os.getenv("BACKEND")
-backend = Client(BACKEND)
 JS_FUNC = os.getenv("JS_FUNC")
+USER = os.getenv("USER")
+PASS = os.getenv("PASS")
+
+backend = Client(BACKEND, auth=[USER, PASS])
 
 def base64_to_image(base64_str):
     return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
@@ -60,7 +63,8 @@ def if_limited(request):
         user_attempts[user_id] = today
     return False
 
-def search_face(file, token, request: gr.Request):
+def search_face(file, token_txt, request: gr.Request):
+    global backend
     try:
         file_1 = handle_file(file)
     except Exception as e:
@@ -71,11 +75,19 @@ def search_face(file, token, request: gr.Request):
         gr.Info("⏳ Wait for your next free search, or 🚀 Go Premium for deep search!", duration=12)
         return []
 
-    result_text = backend.predict(
-            file=file_1,
-            token=token, 
+    try:
+        result_text = backend.predict(
+            file=file1,
+            token=token_txt, 
+            api_name="/search_face"
+        )
+    except:
+        backend = Client(BACKEND, auth=[USER, PASS])
+        result_text = backend.predict(
+            file=file1,
+            token=token_txt, 
             api_name="/search_face"
-    )
+        )
    
     result = json.loads(result_text)
     outarray = []