Upload 3 files

Dockerfile

@@ -0,0 +1,20 @@
+# Base image
+FROM python:3.9-slim
+
+# Working directory set karna
+WORKDIR /code
+
+# Requirements file copy karna
+COPY ./requirements.txt /code/requirements.txt
+
+# Dependencies install karna
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# App code copy karna
+COPY ./app.py /code/app.py
+
+# Port expose karna (Hugging Face ke liye 7860)
+EXPOSE 7860
+
+# Gunicorn se app ko run karna
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,37 @@
+import os
+from flask import Flask, request, jsonify
+from huggingface_hub import InferenceClient
+import base64
+import io
+
+app = Flask(__name__)
+client = InferenceClient(api_key=os.environ.get("HF_TOKEN"))
+
+@app.route('/', methods=['GET'])
+def home():
+    return "Server is running!", 200
+
+@app.route('/generate-image', methods=['POST'])
+def generate_image():
+    prompt = request.json.get('prompt')
+    if not prompt:
+        return jsonify({"error": "Prompt is missing"}), 400
+
+    try:
+        image = client.text_to_image(
+            prompt,
+            model="stabilityai/stable-diffusion-xl-base-1.0",
+        )
+        
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+        return jsonify({"image": img_str})
+
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,4 @@
+flask
+huggingface_hub
+pillow
+gunicorn