gabrielandrade2
commited on
Commit
·
8faec96
1
Parent(s):
0e4a557
Fix -i parameter when input is a directory
Browse files- 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 |
|