feat: use gunicorn

Dockerfile

@@ -8,4 +8,4 @@ RUN chmod -R 777 /app
 
 RUN pip install --break-system-packages --no-cache-dir --requirement requirements.txt
 
-CMD ["python", "app.py"]
+CMD ["gunicorn", "app:app", "--bind", "0.0.0.0:7860", "--worker-class", "gevent"]

app.py

@@ -11,7 +11,7 @@ from diffusers import AutoPipelineForText2Image
 app = Flask("ImageAPI")
 
 
-@app.get("/")
+@app.route(rule="/", methods=["GET"])
 def main():
     prompt = request.args.get("prompt")
 
@@ -34,9 +34,3 @@ def main():
         path_or_file=BytesIO(image),
         mimetype="image/png",
     )
-
-
-app.run(
-    host="localhost",
-    port=7860,
-)

requirements.txt

@@ -1,2 +1,4 @@
 diffusers
-flask
+flask
+gevent
+gunicorn