Update app.py

app.py

@@ -84,8 +84,8 @@ def inference_mask1(
     pred_mask, simi, simi_map = model.predict_mask_nshot(batch, nshot=nshot)
     pred_mask = pred_mask.detach().cpu().numpy()[0]
     output_img = query_img_np.copy()
-    output_img[pred_mask] = np.array([1, 0, 0])
-    output_img = (output_img * 255).astype(np.uint8)
+    output_img[pred_mask] = np.array([255, 0, 0])
+    output_img = (output_img).astype(np.uint8)
     return output_img