sariyam commited on
Commit
1c37c95
Β·
verified Β·
1 Parent(s): 4e70dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,6 +5,7 @@ import spaces
5
  from transformers import AutoModelForImageSegmentation
6
  import torch
7
  from torchvision import transforms
 
8
 
9
  torch.set_float32_matmul_precision(["high", "highest"][0])
10
 
@@ -27,7 +28,7 @@ def fn(image):
27
  origin = im.copy()
28
  processed_image = process(im)
29
 
30
- name_path = "output.png"
31
  processed_image.save(name_path)
32
  return (processed_image , origin) , name_path
33
 
 
5
  from transformers import AutoModelForImageSegmentation
6
  import torch
7
  from torchvision import transforms
8
+ import uuid
9
 
10
  torch.set_float32_matmul_precision(["high", "highest"][0])
11
 
 
28
  origin = im.copy()
29
  processed_image = process(im)
30
 
31
+ name_path = 'bgremove_'+str(uuid.uuid4()) + '.png'
32
  processed_image.save(name_path)
33
  return (processed_image , origin) , name_path
34