feat: first commit

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.12
+
+WORKDIR /app
+
+COPY --chown=1000 . /app
+
+RUN chmod -R 777 /app
+
+RUN pip install --break-system-packages --no-cache-dir --requirement requirements.txt
+
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,42 @@
+from flask import (
+    Flask,
+    jsonify,
+    request,
+    send_file,
+)
+from io import BytesIO
+from diffusers import AutoPipelineForText2Image
+
+
+app = Flask("ImageAPI")
+
+
+@app.get("/")
+def main():
+    prompt = request.args.get("prompt")
+
+    if not prompt:
+        return jsonify({
+            "status": "ERROR",
+            "message": "Prompt is not found!",
+        })
+
+    pipeline = AutoPipelineForText2Image.from_pretrained(pretrained_model_or_path="Lykon/DreamShaper")
+    pipeline.to("cuda")
+    
+    image = pipeline(
+        prompt,
+        height=1024,
+        width=1024,
+    ).images[0]
+
+    return send_file(
+        path_or_file=BytesIO(image),
+        mimetype="image/png",
+    )
+
+
+app.run(
+    host="localhost",
+    port=7860,
+)

requirements.txt

@@ -0,0 +1 @@
+flask