Update app.py

app.py

@@ -53,20 +53,18 @@ import PyTorch
 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):
-    #     # Your existing PyTorch conversion  
-    if isinstance(image, np.ndarray): 
+def process_image_for_pil(image): 
+    if isinstance(image, torch.Tensor):
+        # 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.")
 
 
 
+
+
 def should_resize():
     """Determines whether to resize images (replace with your own logic)"""
     # Example: Resize only if the image dimensions exceed a threshold