sourav520 commited on
Commit
f59bc45
·
verified ·
1 Parent(s): 2cad0ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -30,12 +30,12 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
30
  # Prepare saved models folder
31
  if not os.path.exists("saved_models"):
32
  os.mkdir("saved_models")
33
- # NOTE: make sure isnet.pth is available, otherwise manual download needed
34
  os.system("mv isnet.pth saved_models/")
35
 
36
  # --- Helpers ---
37
  class GOSNormalize(object):
38
- def __init__(self, mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225]):
39
  self.mean = mean
40
  self.std = std
41
 
@@ -43,7 +43,7 @@ class GOSNormalize(object):
43
  return normalize(image, self.mean, self.std)
44
 
45
  transform = transforms.Compose([
46
- GOSNormalize([0.5,0.5,0.5],[1.0,1.0,1.0])
47
  ])
48
 
49
  def load_image(im_path, hypar):
@@ -134,12 +134,12 @@ interface = gr.Interface(
134
  gr.Image(type='filepath', format="png", visible=False)
135
  ],
136
  flagging_mode="never",
137
- cache_mode="lazy"
 
138
  )
139
 
140
  interface.launch(
141
  show_error=False,
142
  show_api=False,
143
- share=False,
144
- css=css_hide_footer
145
  )
 
30
  # Prepare saved models folder
31
  if not os.path.exists("saved_models"):
32
  os.mkdir("saved_models")
33
+ # NOTE: make sure isnet.pth is available locally
34
  os.system("mv isnet.pth saved_models/")
35
 
36
  # --- Helpers ---
37
  class GOSNormalize(object):
38
+ def __init__(self, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]):
39
  self.mean = mean
40
  self.std = std
41
 
 
43
  return normalize(image, self.mean, self.std)
44
 
45
  transform = transforms.Compose([
46
+ GOSNormalize([0.5, 0.5, 0.5], [1.0, 1.0, 1.0])
47
  ])
48
 
49
  def load_image(im_path, hypar):
 
134
  gr.Image(type='filepath', format="png", visible=False)
135
  ],
136
  flagging_mode="never",
137
+ cache_mode="lazy",
138
+ css=css_hide_footer # ✅ CSS here inside Interface, not launch
139
  )
140
 
141
  interface.launch(
142
  show_error=False,
143
  show_api=False,
144
+ share=False
 
145
  )