Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,11 +26,11 @@ def main():
|
|
26 |
|
27 |
# DL MODEL
|
28 |
# PIX_MODEL
|
29 |
-
os.environ['PIX_MODEL'] = huggingface_hub.hf_hub_download("
|
30 |
# NET_MODEL
|
31 |
-
os.environ['NET_MODEL'] = huggingface_hub.hf_hub_download("
|
32 |
# ALIAS_MODEL
|
33 |
-
os.environ['ALIAS_MODEL'] = huggingface_hub.hf_hub_download("
|
34 |
|
35 |
# For local testing
|
36 |
# PIX_MODEL
|
@@ -51,7 +51,7 @@ def main():
|
|
51 |
gr.Interface(m.pixelize_modified,
|
52 |
[
|
53 |
gr.components.Image(type='pil', label='Input'),
|
54 |
-
gr.components.Slider(minimum=
|
55 |
gr.components.Checkbox(True, label="Upscale after")
|
56 |
],
|
57 |
gr.components.Image(type='pil', label='Output'),
|
|
|
26 |
|
27 |
# DL MODEL
|
28 |
# PIX_MODEL
|
29 |
+
os.environ['PIX_MODEL'] = huggingface_hub.hf_hub_download("Ferdian/pixelization_models", "pixelart_vgg19.pth", token=TOKEN);
|
30 |
# NET_MODEL
|
31 |
+
os.environ['NET_MODEL'] = huggingface_hub.hf_hub_download("Ferdian/pixelization_models", "160_net_G_A.pth", token=TOKEN);
|
32 |
# ALIAS_MODEL
|
33 |
+
os.environ['ALIAS_MODEL'] = huggingface_hub.hf_hub_download("Ferdian/pixelization_models", "alias_net.pth", token=TOKEN);
|
34 |
|
35 |
# For local testing
|
36 |
# PIX_MODEL
|
|
|
51 |
gr.Interface(m.pixelize_modified,
|
52 |
[
|
53 |
gr.components.Image(type='pil', label='Input'),
|
54 |
+
gr.components.Slider(minimum=1, maximum=32, value=1, step=1, label='Pixel Size'),
|
55 |
gr.components.Checkbox(True, label="Upscale after")
|
56 |
],
|
57 |
gr.components.Image(type='pil', label='Output'),
|