Spaces:
Runtime error
Runtime error
Update main.py
Browse files
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 |
-
|
61 |
if not auth_header.startswith('Bearer ' + apiKey):
|
62 |
-
return jsonify({
|
|
|
|
|
|
|
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 |
}
|