Update app.py

app.py

@@ -47,6 +47,7 @@ from diffusers import ShapEPipeline
 from diffusers.utils import export_to_gif
 from PIL import Image
 import numpy as np
+import PyTorch
 
 # Model loading (Ideally done once at the start for efficiency)
 ckpt_id = "openai/shap-e"  
@@ -54,8 +55,13 @@ ckpt_id = "openai/shap-e"
 @st.cache_resource  # Caches the model for faster subsequent runs
 def process_image_for_pil(image):
     if isinstance(image, torch.Tensor):
-        image_array = image.detach().cpu().numpy()
-        return Image.fromarray(image_array)  
+        # Your existing PyTorch conversion  
+    elif isinstance(image, np.ndarray): 
+        # Your existing Numpy conversion
+    elif isinstance(image, list):  # Assuming a nested list structure
+        flattened_data = [pixel for sublist in image for pixel in sublist]
+        image_array = np.array(flattened_data).reshape(256, 256, 3)  # Hypothetical reshape 
+        return Image.fromarray(image_array) 
     else: 
         raise TypeError("Unsupported image format. Please provide conversion logic.")