Update app.py

app.py

@@ -355,7 +355,7 @@ def inference(input_prompt, input_category):
                         # cv2.imwrite(os.path.join("demo/demo_mask.png"), done_image_mask)
                         
                         # torchvision.utils.save_image(annotation_pred, os.path.join("demo/demo_segresult.png"), normalize=True, scale_each=True)
-                        generated_image = x_sample
+                        generated_image = x_sample.astype(np.uint8)
                         generated_mask = done_image_mask
                         return [generated_image, generated_mask]