Update app.py

app.py

@@ -51,6 +51,33 @@ from PIL import Image
 ckpt_id = "openai/shap-e"  
 
 @st.cache_resource  # Caches the model for faster subsequent runs
+def process_image_for_pil(image):
+    """Converts image data into a format compatible with PIL.Image.fromarray()""" 
+
+    # Example 1: If image is already a PIL Image
+    if isinstance(image, Image.Image):
+        return image 
+
+    # Example 2: If image is a NumPy array
+    if isinstance(image, np.ndarray):
+        return Image.fromarray(image)  
+
+    # Example 3: If image is a PyTorch Tensor
+    if isinstance(image, torch.Tensor):
+        return Image.fromarray(image.detach().cpu().numpy()) 
+
+    # Add more cases based on the output of print(type(images[0])) 
+
+    raise TypeError("Unsupported image format. Please provide conversion logic.")
+
+
+def should_resize():
+    """Determines whether to resize images (replace with your own logic)"""
+    # Example: Resize only if the image dimensions exceed a threshold
+    if image.width > 512 or image.height > 512: 
+        return True
+    else:
+        return False 
 def load_model():    
     return ShapEPipeline.from_pretrained(ckpt_id).to("cuda")