Update app.py

app.py

@@ -45,7 +45,7 @@ config.blip_num_beams = 64
 
 ci = Interrogator(config)
 
-import numpy as np
+from PIL import Image
 
 def inference(input_images, mode, best_max_flavors):
     processed_images = []  # To store processed images
@@ -64,8 +64,7 @@ def inference(input_images, mode, best_max_flavors):
         prompt_results.append(prompt_result)
 
         # Append processed image data to the list
-        processed_image = np.array(image)  # Convert PIL image to numpy array
-        processed_images.append(processed_image)
+        processed_images.append(image)  # Append PIL image directly
 
     return processed_images, prompt_results  # Return both the processed images and prompt results