Update app.py

app.py

@@ -52,23 +52,13 @@ 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()""" 
+    """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])) 
+    if isinstance(image, np.ndarray):  # Assuming NumPy array
+        return Image.fromarray(image)
+    else:
+        raise TypeError("Unsupported image format. Please provide conversion logic.")
 
-    raise TypeError("Unsupported image format. Please provide conversion logic.")
 
 
 def should_resize():