Image Classification
timm
drhead commited on
Commit
04410b2
1 Parent(s): 8449d61

Update JTP_PILOT2/inference_gradio.py

Browse files
Files changed (1) hide show
  1. JTP_PILOT2/inference_gradio.py +1 -1
JTP_PILOT2/inference_gradio.py CHANGED
@@ -145,7 +145,7 @@ if torch.cuda.is_available():
145
 
146
  model.eval()
147
 
148
- with open("tagger_tags.json", "r") as file:
149
  tags = json.load(file) # type: dict
150
  allowed_tags = list(tags.keys())
151
 
 
145
 
146
  model.eval()
147
 
148
+ with open("tags.json", "r") as file:
149
  tags = json.load(file) # type: dict
150
  allowed_tags = list(tags.keys())
151