Update app.py

app.py

@@ -1,7 +1,11 @@
 import os
 import logging
 from flask import Flask, request, jsonify, send_file
-from diffusers import DiffusionPipeline
+from diffusers.pipelines import DiffusionPipeline
+try:
+    from diffusers.pipelines.zero123plus.pipeline_zero123plus import Zero123PlusPipeline
+except ImportError:
+    print("Zero123PlusPipeline not found in current diffusers version")
 import torch
 from PIL import Image
 import io
@@ -16,14 +20,13 @@ logger = logging.getLogger(__name__)
 
 # Set Hugging Face cache directory to a writable path
 # Make sure to set this BEFORE importing or initializing any models
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
 os.environ['HF_HOME'] = '/tmp/hf_home'
 os.environ['XDG_CACHE_HOME'] = '/tmp/cache'
 
 # Create cache directories if they don't exist
-os.makedirs('/tmp/transformers_cache', exist_ok=True)
 os.makedirs('/tmp/hf_home', exist_ok=True)
 os.makedirs('/tmp/cache', exist_ok=True)
+os.makedirs('/tmp/diffusers_cache', exist_ok=True)
 
 # Global variable for the model
 pipe = None
@@ -33,10 +36,13 @@ def load_model():
     global pipe
     try:
         logger.info("Loading Zero123Plus model...")
+        # Use auto_mapping to let diffusers figure out the correct pipeline class
         pipe = DiffusionPipeline.from_pretrained(
             "sudo-ai/zero123plus-v1.2",
             torch_dtype=torch.float32,
-            cache_dir="/tmp/diffusers_cache"
+            cache_dir="/tmp/diffusers_cache",
+            local_files_only=False,
+            resume_download=True
         )
         pipe.to("cpu")
         logger.info("Model loaded successfully")