Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ try:
|
|
27 |
if token != API_KEY:
|
28 |
raise HTTPException(status_code=401, detail="Unauthorized")
|
29 |
try:
|
30 |
-
output = pipe(request.prompt
|
31 |
return GenerateResponse(generated_text=output)
|
32 |
except Exception as e:
|
33 |
raise HTTPException(status_code=500, detail=str(e))
|
|
|
27 |
if token != API_KEY:
|
28 |
raise HTTPException(status_code=401, detail="Unauthorized")
|
29 |
try:
|
30 |
+
output = pipe(request.prompt)[0]['generated_text']
|
31 |
return GenerateResponse(generated_text=output)
|
32 |
except Exception as e:
|
33 |
raise HTTPException(status_code=500, detail=str(e))
|