Update app.py

app.py

@@ -31,6 +31,7 @@ try:
 
     class GenerateResponse(BaseModel):
         generated_text: str
+        usage: TokenUsage
 
     @app.post("/generate", response_model=GenerateResponse)
     async def generate(request: GenerateRequest):