Update app.py

app.py

@@ -15,7 +15,7 @@ from huggingface_hub import snapshot_download
 from flask_cors import CORS
 import numpy as np
 import trimesh
-from diffusers import Hunyuan3DDiTPipeline
+from diffusers import StableFast3DPipeline
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 torch.set_default_device("cpu")
@@ -42,12 +42,12 @@ app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
 
 processing_jobs = {}
-hunyuan_pipeline = None
+sf3d_pipeline = None
 model_loaded = False
 model_loading = False
 
 TIMEOUT_SECONDS = 300
-MAX_DIMENSION = 512  # Hunyuan3D-1.0 uses 512x512 inputs
+MAX_DIMENSION = 512  # Stable-Fast-3D uses 512x512 inputs
 
 class TimeoutError(Exception):
     pass
@@ -94,21 +94,21 @@ def preprocess_image(image_path):
         raise Exception(f"Error preprocessing image: {str(e)}")
 
 def load_model():
-    global hunyuan_pipeline, model_loaded, model_loading
+    global sf3d_pipeline, model_loaded, model_loading
     
     if model_loaded:
-        return hunyuan_pipeline
+        return sf3d_pipeline
     
     if model_loading:
         while model_loading and not model_loaded:
             time.sleep(0.5)
-        return hunyuan_pipeline
+        return sf3d_pipeline
     
     try:
         model_loading = True
-        print("Loading Hunyuan3D-1.0 Lite...")
+        print("Loading Stable-Fast-3D...")
         
-        model_name = "tencent/Hunyuan3D-1"
+        model_name = "stabilityai/stable-fast-3d"
         
         max_retries = 3
         retry_delay = 5
@@ -128,17 +128,16 @@ def load_model():
                 else:
                     raise
         
-        hunyuan_pipeline = Hunyuan3DDiTPipeline.from_pretrained(
+        sf3d_pipeline = StableFast3DPipeline.from_pretrained(
             model_name,
-            subfolder="lite",
             cache_dir=CACHE_DIR,
             torch_dtype=torch.float32,
         )
-        hunyuan_pipeline.to("cpu")
+        sf3d_pipeline.to("cpu")
         
         model_loaded = True
-        print("Hunyuan3D-1.0 Lite loaded successfully on CPU")
-        return hunyuan_pipeline
+        print("Stable-Fast-3D loaded successfully on CPU")
+        return sf3d_pipeline
     
     except Exception as e:
         print(f"Error loading model: {str(e)}")
@@ -153,12 +152,17 @@ def generate_3d_model(image, detail_level):
         steps = num_steps[detail_level]
         
         with torch.no_grad():
-            result = hunyuan_pipeline(image, num_inference_steps=steps, max_faces_num=20000)
+            result = sf3d_pipeline(
+                image,
+                num_inference_steps=steps,
+                normal_num_inference_steps=steps // 2,
+                guidance_scale=7.0,
+            )
         
-        mesh = result[0]  # Hunyuan3D returns a trimesh object
+        mesh = result.trimesh_meshes[0]
         vertices = np.array(mesh.vertices)
         faces = np.array(mesh.faces)
-        vertex_colors = np.array(mesh.vertex_colors) if mesh.vertex_colors is not None else None
+        vertex_colors = np.array(mesh.visual.vertex_colors) if mesh.visual.vertex_colors is not None else None
         
         trimesh_mesh = trimesh.Trimesh(
             vertices=vertices,
@@ -176,7 +180,7 @@ def generate_3d_model(image, detail_level):
 def health_check():
     return jsonify({
         "status": "healthy",
-        "model": "Hunyuan3D-1.0 Lite",
+        "model": "Stable-Fast-3D",
         "device": "cpu"
     }), 200
 
@@ -355,7 +359,7 @@ def preview_model(job_id):
         else:
             return send_file(file_path, mimetype='text/plain')
     
-    return jsonify({"error": "Model file not found"}), 404
+    return jsonify({"error": "Model not found"}), 404
 
 def cleanup_old_jobs():
     current_time = time.time()
@@ -416,7 +420,7 @@ def model_info(job_id):
 @app.route('/', methods=['GET'])
 def index():
     return jsonify({
-        "message": "Image to 3D API (Hunyuan3D-1.0 Lite)",
+        "message": "Image to 3D API (Stable-Fast-3D)",
         "endpoints": [
             "/convert",
             "/progress/<job_id>",
@@ -428,7 +432,7 @@ def index():
             "output_format": "glb or obj",
             "detail_level": "low, medium, or high"
         },
-        "description": "Creates 3D models from 2D images using Hunyuan3D-1.0 Lite."
+        "description": "Creates 3D models from 2D images using Stable-Fast-3D."
     }), 200
 
 if __name__ == '__main__':