Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,8 +23,8 @@ try:
|
|
23 |
@app.post("/generate", response_model=GenerateResponse)
|
24 |
async def generate(request: GenerateRequest):
|
25 |
authorization = request.key
|
26 |
-
token = authorization.split('Bearer ')[1]
|
27 |
-
if
|
28 |
raise HTTPException(status_code=401, detail="Unauthorized")
|
29 |
try:
|
30 |
output = pipe(request.prompt)[0]['generated_text']
|
|
|
23 |
@app.post("/generate", response_model=GenerateResponse)
|
24 |
async def generate(request: GenerateRequest):
|
25 |
authorization = request.key
|
26 |
+
#token = authorization.split('Bearer ')[1]
|
27 |
+
if authorization != API_KEY:
|
28 |
raise HTTPException(status_code=401, detail="Unauthorized")
|
29 |
try:
|
30 |
output = pipe(request.prompt)[0]['generated_text']
|