fix: dev to prod

app.py

@@ -6,7 +6,7 @@ import argparse
 import huggingface_hub
 import os
 
-# TOKEN = os.environ['TOKEN']
+TOKEN = os.environ['TOKEN']
 
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
@@ -26,19 +26,19 @@ def main():
 
   # DL MODEL
   # PIX_MODEL
-  # os.environ['PIX_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "pixelart_vgg19.pth", token=TOKEN);
-  # # NET_MODEL
-  # os.environ['NET_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "160_net_G_A.pth", token=TOKEN);
-  # # ALIAS_MODEL
-  # os.environ['ALIAS_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "alias_net.pth", token=TOKEN);
+  os.environ['PIX_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "pixelart_vgg19.pth", token=TOKEN);
+  # NET_MODEL
+  os.environ['NET_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "160_net_G_A.pth", token=TOKEN);
+  # ALIAS_MODEL
+  os.environ['ALIAS_MODEL'] = huggingface_hub.hf_hub_download("NoCrypt/pixelization_models", "alias_net.pth", token=TOKEN);
   
   # For local testing
   # PIX_MODEL
-  os.environ['PIX_MODEL'] = "pixelart_vgg19.pth"
-  # NET_MODEL
-  os.environ['NET_MODEL'] = "160_net_G_A.pth"
-  # ALIAS_MODEL
-  os.environ['ALIAS_MODEL'] = "alias_net.pth"
+  # os.environ['PIX_MODEL'] = "pixelart_vgg19.pth"
+  # # NET_MODEL
+  # os.environ['NET_MODEL'] = "160_net_G_A.pth"
+  # # ALIAS_MODEL
+  # os.environ['ALIAS_MODEL'] = "alias_net.pth"
 
 
   use_cpu = True