Update app.py

app.py

@@ -20,10 +20,10 @@ os.environ['HF_HOME'] = '/tmp/.cache/huggingface'
 # Create cache directory if it doesn't exist
 os.makedirs('/tmp/.cache/huggingface', exist_ok=True)
 
-app = Flask(__name__)
+# Global variable for the model
+pipe = None
 
-# Initialize the model
-@app.before_first_request
+# Initialize the model at startup
 def load_model():
     global pipe
     try:
@@ -38,6 +38,11 @@ def load_model():
         logger.error(f"Error loading model: {str(e)}")
         raise
 
+# Load the model immediately
+load_model()
+
+app = Flask(__name__)
+
 @app.route("/", methods=["GET"])
 def index():
     return jsonify({"message": "Zero123Plus API is running."})
@@ -45,6 +50,15 @@ def index():
 @app.route("/generate", methods=["POST"])
 def generate():
     try:
+        global pipe
+        # Check if model is loaded
+        if pipe is None:
+            try:
+                load_model()
+            except Exception as e:
+                logger.error(f"Failed to load model: {str(e)}")
+                return jsonify({"error": "Failed to initialize model"}), 500
+        
         if 'image' not in request.files:
             logger.warning("No image uploaded")
             return jsonify({"error": "No image uploaded"}), 400