fix pydantic

main.py

@@ -69,13 +69,13 @@ def generate(request: GenerationRequest):
         if output[0]:
             request = output[1]
         else:
-            return {'error': output[1]}
+            return ErrorOutput(error=output[1])
         
         images = model.sample(request)
         process_time = time.perf_counter() - t
         logger.info(f"Request took {process_time:0.3f} seconds")
         print(images)
-        return {"generations": images}
+        return GenerationRequest(generation=images)
 
     except Exception as e:
         capture_exception(e)