output instead of generation

main.py

@@ -65,7 +65,7 @@ class GenerationRequest(BaseModel):
     seed: int = None
 
 class GenerationOutput(BaseModel):
-    generation: List[str]
+    output: List[str]
 
 class ErrorOutput(BaseModel):
     error: str
@@ -91,7 +91,7 @@ def generate(request: GenerationRequest):
 
         process_time = time.perf_counter() - t
         logger.info(f"Request took {process_time:0.3f} seconds")
-        return GenerationOutput(generation=images)
+        return GenerationOutput(output=images)
 
     except Exception as e:
         capture_exception(e)