petertill commited on
Commit
14cdd4c
·
verified ·
1 Parent(s): 67e4952

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 token != API_KEY:
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']