akhaliq HF staff commited on
Commit
3fa910c
1 Parent(s): 811f01f

one more fix

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -13,6 +13,10 @@ from timm.data import create_transform
13
  from config import get_config
14
  from model import build_model
15
 
 
 
 
 
16
  def parse_option():
17
  parser = argparse.ArgumentParser('UniCL demo script', add_help=False)
18
  parser.add_argument('--cfg', type=str, default="configs/unicl_swin_base.yaml", metavar="FILE", help='path to config file', )
 
13
  from config import get_config
14
  from model import build_model
15
 
16
+ # Download human-readable labels for ImageNet.
17
+ response = requests.get("https://git.io/JJkYN")
18
+ labels = response.text.split("\n")
19
+
20
  def parse_option():
21
  parser = argparse.ArgumentParser('UniCL demo script', add_help=False)
22
  parser.add_argument('--cfg', type=str, default="configs/unicl_swin_base.yaml", metavar="FILE", help='path to config file', )