Update app.py

app.py

@@ -1,40 +1,32 @@
-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)
+import os from flask import Flask, request, jsonify, send_file from diffusers import DiffusionPipeline import torch from PIL import Image import io
+
+Set Hugging Face cache directory to a writable path
+
+os.environ['HF_HOME'] = '/home/user/.cache/huggingface'
+
+app = Flask(name)
+
+Load the model on CPU
+
+pipe = DiffusionPipeline.from_pretrained( "sudo-ai/zero123plus-v1.2", torch_dtype=torch.float32 ) pipe.to("cpu")
+
+@app.route("/", methods=["GET"]) def index(): return jsonify({"message": "Zero123Plus API is running."})
+
+@app.route("/generate", methods=["POST"]) def generate(): if 'image' not in request.files: return jsonify({"error": "No image uploaded"}), 400
+
+image_file = request.files['image']
+input_image = Image.open(image_file).convert("RGB")
+
+# Generate new views
+result = pipe(image=input_image, num_inference_steps=25)
+output_image = result.images[0]
+
+# Save to a BytesIO object
+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)
+