Ferdian commited on
Commit
d049c57
·
verified ·
1 Parent(s): 47688b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("NoCrypt/pixelization_models", "pixelart_vgg19.pth", token=TOKEN);
30
  # NET_MODEL
31
- os.environ['NET_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "160_net_G_A.pth", token=TOKEN);
32
  # ALIAS_MODEL
33
- os.environ['ALIAS_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "alias_net.pth", token=TOKEN);
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=4, maximum=32, value=4, step=1, label='Pixel Size'),
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'),