Update app.py

app.py

@@ -6,11 +6,16 @@ import requests
 from io import BytesIO
 
 # Load model and processor from Hugging Face
-try:
-    model = AutoModel.from_pretrained("zxhezexin/openlrm-mix-large-1.1")
-    processor = AutoProcessor.from_pretrained("zxhezexin/openlrm-mix-large-1.1")
-except Exception as e:
-    print(f"Error loading model or processor: {e}")
+def load_model_and_processor():
+    try:
+        model = AutoModel.from_pretrained("zxhezexin/openlrm-mix-large-1.1")
+        processor = AutoProcessor.from_pretrained("zxhezexin/openlrm-mix-large-1.1")
+        return model, processor
+    except Exception as e:
+        print(f"Error loading model or processor: {e}")
+        return None, None
+
+model, processor = load_model_and_processor()
 
 # Example image URL (replace this with a suitable example)
 example_image_url = "https://huggingface.co/datasets/nateraw/image-folder/resolve/main/example_1.png"
@@ -27,6 +32,9 @@ def load_example_image():
 
 # Define function to generate 3D output from 2D image
 def image_to_3d(image):
+    if processor is None or model is None:
+        return "Model or processor not loaded."
+    
     try:
         # Preprocess the input image
         inputs = processor(images=image, return_tensors="pt")