huedaya commited on
Commit
dc7b6bf
·
verified ·
1 Parent(s): b489664

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -4
main.py CHANGED
@@ -57,11 +57,13 @@ def generate(item: Item):
57
  async def generate_text(item: Item):
58
  # Reject if not authenticated
59
  apiKey = os.environ.get("API_KEY")
60
- auth_header = request.headers['Authorization']
61
  if not auth_header.startswith('Bearer ' + apiKey):
62
- return jsonify({'message': 'Invalid or missing Bearer token'}), 401
 
 
 
63
 
64
  return {
65
- "response": generate(item),
66
- "token": auth_header
67
  }
 
57
  async def generate_text(item: Item):
58
  # Reject if not authenticated
59
  apiKey = os.environ.get("API_KEY")
60
+ authHeader = requests.headers['Authorization']
61
  if not auth_header.startswith('Bearer ' + apiKey):
62
+ return jsonify({
63
+ "message": "Invalid or missing Bearer token",
64
+ "token": authHeader
65
+ }), 401
66
 
67
  return {
68
+ "response": generate(item)
 
69
  }