gabrielandrade2 commited on
Commit
8faec96
1 Parent(s): 0e4a557

Fix -i parameter when input is a directory

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -150,7 +150,7 @@ def run(model, input, output=None, normalize=False, **kwargs):
150
  id_to_tags = pickle.load(tf)
151
 
152
  if (os.path.isdir(input)):
153
- files = [f for f in os.listdir(input) if os.path.isfile(os.path.join(input, f))]
154
  else:
155
  files = [input]
156
 
 
150
  id_to_tags = pickle.load(tf)
151
 
152
  if (os.path.isdir(input)):
153
+ files = [os.path.join(input, f) for f in os.listdir(input) if os.path.isfile(os.path.join(input, f))]
154
  else:
155
  files = [input]
156