Update app.py

app.py

@@ -1,32 +1,44 @@
-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
-
+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 = Flask(__name__)
 
-@app.route("/", methods=["GET"]) def index(): return jsonify({"message": "Zero123Plus API is running."})
+# Load the model on CPU
+pipe = DiffusionPipeline.from_pretrained(
+    "sudo-ai/zero123plus-v1.2",
+    torch_dtype=torch.float32
+)
+pipe.to("cpu")
 
-@app.route("/generate", methods=["POST"]) def generate(): if 'image' not in request.files: return jsonify({"error": "No image uploaded"}), 400
+@app.route("/", methods=["GET"])
+def index():
+    return jsonify({"message": "Zero123Plus API is running."})
 
-image_file = request.files['image']
-input_image = Image.open(image_file).convert("RGB")
+@app.route("/generate", methods=["POST"])
+def generate():
+    if 'image' not in request.files:
+        return jsonify({"error": "No image uploaded"}), 400
 
-# Generate new views
-result = pipe(image=input_image, num_inference_steps=25)
-output_image = result.images[0]
+    image_file = request.files['image']
+    input_image = Image.open(image_file).convert("RGB")
 
-# Save to a BytesIO object
-img_io = io.BytesIO()
-output_image.save(img_io, 'PNG')
-img_io.seek(0)
+    # Generate new views
+    result = pipe(image=input_image, num_inference_steps=25)
+    output_image = result.images[0]
 
-return send_file(img_io, mimetype='image/png')
+    # Save to a BytesIO object
+    img_io = io.BytesIO()
+    output_image.save(img_io, 'PNG')
+    img_io.seek(0)
 
-if name == "main": app.run(host="0.0.0.0", port=7860)
+    return send_file(img_io, mimetype='image/png')
 
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)