Samarth991 commited on
Commit
90b20c2
1 Parent(s): a606ba4

updated preds

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,7 +4,9 @@ from transformers import pipeline
4
  import numpy as np
5
  from PIL import Image
6
  import torch
 
7
  import cv2
 
8
  from matplotlib import pyplot as plt
9
  from segmentation_mask_overlay import overlay_masks
10
  from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation,AutoProcessor,AutoConfig
 
4
  import numpy as np
5
  from PIL import Image
6
  import torch
7
+ from torch import nn
8
  import cv2
9
+
10
  from matplotlib import pyplot as plt
11
  from segmentation_mask_overlay import overlay_masks
12
  from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation,AutoProcessor,AutoConfig