Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import torch
 import time
@@ -20,6 +19,10 @@ import cv2
 import pymeshlab
 from hy3dgen.shapegen import Hunyuan3DDiTFlowMatchingPipeline
 
+# Force CPU usage
+os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disable GPU detection
+torch.set_default_device("cpu")  # Set CPU as default device
+
 app = Flask(__name__)
 CORS(app)  # Enable CORS for all routes
 
@@ -150,12 +153,10 @@ def load_model():
             subfolder="hunyuan3d-dit-v2-mini",
             use_safetensors=True,
             torch_dtype=torch.float16,
-            cache_dir=CACHE_DIR
+            cache_dir=CACHE_DIR,
+            device_map="cpu"  # Explicitly set to CPU
         )
         
-        # Move to CPU
-        hunyuan_pipeline.to("cpu")
-        
         model_loaded = True
         print("Model loaded successfully on CPU")
         return hunyuan_pipeline
@@ -448,4 +449,4 @@ def index():
 if __name__ == '__main__':
     cleanup_old_jobs()
     port = int(os.environ.get('PORT', 7860))
-    app.run(host='0.0.0.0', port=port)
+    app.run(host='0.0.0.0', port=port)