Create app.py

app.py

@@ -0,0 +1,40 @@
+from flask import Flask, request, jsonify, send_file
+import torch
+from diffusers import DiffusionPipeline
+from PIL import Image
+import io
+import os
+
+app = Flask(__name__)
+
+# Load the model once when the app starts
+pipe = DiffusionPipeline.from_pretrained(
+    "ashawkey/zero123-xl-diffusers",
+    torch_dtype=torch.float32
+)
+pipe.to("cpu")  # or "cuda" if you have GPU
+
+@app.route('/')
+def index():
+    return jsonify({"status": "Zero123 API is running"})
+
+@app.route('/generate', methods=['POST'])
+def generate():
+    if 'image' not in request.files:
+        return jsonify({"error": "No image uploaded"}), 400
+    
+    file = request.files['image']
+    input_image = Image.open(file).convert("RGB")
+
+    output = pipe(input_image, num_inference_steps=75)
+    output_image = output.images[0]
+
+    # Save image to a buffer
+    img_io = io.BytesIO()
+    output_image.save(img_io, 'PNG')
+    img_io.seek(0)
+    
+    return send_file(img_io, mimetype='image/png')
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)