Revert "Print correct filenames when files are skipped"
This reverts commit b0b93eb471.
Broke the webapp output.
This commit is contained in:
@@ -35,14 +35,12 @@ class Autotagger:
|
||||
except:
|
||||
print("skipped file " + file.name, file=sys.stderr)
|
||||
return None
|
||||
images = [create_image(file) for file in files]
|
||||
files = [files[i] for i in range(len(files)) if images[i] != None]
|
||||
images = [image for image in images if image != None]
|
||||
images = list(filter(lambda i: i != None, [create_image(file) for file in files]))
|
||||
dl = self.learn.dls.test_dl(images, bs=bs)
|
||||
batch, _ = self.learn.get_preds(dl=dl)
|
||||
|
||||
for scores, f in zip(batch, files):
|
||||
for scores in batch:
|
||||
df = DataFrame({ "tag": self.learn.dls.vocab, "score": scores })
|
||||
df = df[df.score >= threshold].sort_values("score", ascending=False).head(limit)
|
||||
tags = dict(zip(df.tag, df.score))
|
||||
yield f.name, tags
|
||||
yield tags
|
||||
|
||||
Reference in New Issue
Block a user