Update app.py
Browse files
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
|
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 |
)
|