Image Classification
timm
drhead commited on
Commit
3b61449
1 Parent(s): 04410b2

Update JTP_PILOT/inference_gradio.py

Browse files
Files changed (1) hide show
  1. JTP_PILOT/inference_gradio.py +1 -1
JTP_PILOT/inference_gradio.py CHANGED
@@ -126,7 +126,7 @@ if torch.cuda.is_available():
126
 
127
  model.eval()
128
 
129
- with open("tagger_tags.json", "r") as file:
130
  tags = json.load(file) # type: dict
131
  allowed_tags = list(tags.keys())
132
 
 
126
 
127
  model.eval()
128
 
129
+ with open("tags.json", "r") as file:
130
  tags = json.load(file) # type: dict
131
  allowed_tags = list(tags.keys())
132