Yao211 commited on
Commit
cde4007
·
verified ·
1 Parent(s): 442677a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -6,6 +6,7 @@ import json
6
 
7
  # Configuration
8
  API_BASE_URL = os.getenv("API_BASE_URL")
 
9
 
10
  def face_compare(frame1, frame2, request: gr.Request = None):
11
  """Face comparison with enhanced result display"""
@@ -22,8 +23,13 @@ def face_compare(frame1, frame2, request: gr.Request = None):
22
  if not files:
23
  return "<div class='error-message'>Please upload both images</div>"
24
 
 
 
 
 
 
25
  # Make API request
26
- response = requests.post(url=url, files=files)
27
  result = response.json()
28
 
29
  # Close files
@@ -34,7 +40,7 @@ def face_compare(frame1, frame2, request: gr.Request = None):
34
  return format_face_comparison_result(result, frame1, frame2)
35
 
36
  except Exception as e:
37
- return f"<div class='error-message'>Error processing request: {str(e)}</div>"
38
 
39
  def format_face_comparison_result(result, img1_path, img2_path):
40
  """Format face comparison results with professional styling"""
 
6
 
7
  # Configuration
8
  API_BASE_URL = os.getenv("API_BASE_URL")
9
+ API_TOKEN = os.getenv("API_TOKEN")
10
 
11
  def face_compare(frame1, frame2, request: gr.Request = None):
12
  """Face comparison with enhanced result display"""
 
23
  if not files:
24
  return "<div class='error-message'>Please upload both images</div>"
25
 
26
+ # Add Bearer token to headers
27
+ headers = {
28
+ "Authorization": f"Bearer {API_TOKEN}"
29
+ }
30
+
31
  # Make API request
32
+ response = requests.post(url=url, files=files, headers=headers)
33
  result = response.json()
34
 
35
  # Close files
 
40
  return format_face_comparison_result(result, frame1, frame2)
41
 
42
  except Exception as e:
43
+ return f"<div class='error-message'>Error processing request</div>"
44
 
45
  def format_face_comparison_result(result, img1_path, img2_path):
46
  """Format face comparison results with professional styling"""