Changes

app.py

@@ -85,7 +85,7 @@ else:
 
 
 def process_image(input_image):
-    image = Image.open(io.BytesIO(input_image)).convert("RGB")
+    image = Image.fromarray(input_image).convert("RGB")
 
     start_time = time.time()
     heatmap = scanmap(np.array(image), model)
@@ -96,6 +96,7 @@ def process_image(input_image):
 
     return image, heatmap_img, int(elapsed_time)
 
+
 def scanmap(image_np, model):
     image_np = image_np.astype(np.float32) / 255.0
 
@@ -123,7 +124,7 @@ def scanmap(image_np, model):
     return probabilities_map
 
 def gradio_process_image(input_image):
-    original, heatmap, elapsed_time = process_image(input_image.read())
+    original, heatmap, elapsed_time = process_image(input_image)
     return original, heatmap, f"Elapsed Time (seconds): {elapsed_time}"
 
 inputs = gr.Image(label="Upload Image")