Update app.py

app.py

@@ -271,6 +271,9 @@ def segment_image(image):
     return image, legend, stats, superimposed_image
 
 def superimpose_images(image1, image2, alpha=0.5):
+
+    if image1.dtype != np.uint8:
+            image1 = (image1 * 255).astype(np.uint8) if image1.max() <= 1 else image1.astype(np.uint8)
     # Convert NumPy arrays to PIL images
     image1 = Image.fromarray(image1)