a

Dockerfile

@@ -35,5 +35,5 @@ EXPOSE 50051
 WORKDIR /usr/src/app
 COPY . .
 
-#CMD [ "gunicorn", "main:app --workers 1 --worker-class uvicorn.workers.UvicornWorker --bind 0.0.0.0:80" ]
-CMD [ "python3", "main.py" ]
\ No newline at end of file
+CMD [ "gunicorn", "main:app --workers 1 --worker-class uvicorn.workers.UvicornWorker --bind 0.0.0.0:80" ]
+#CMD [ "python3", "main.py" ]
\ No newline at end of file

main.py

@@ -74,6 +74,7 @@ def generate(request: GenerationRequest):
         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}
 
     except Exception as e: