diff --git a/src/Metalhead.jl b/src/Metalhead.jl index 9785b10fe..194b32f86 100644 --- a/src/Metalhead.jl +++ b/src/Metalhead.jl @@ -4,7 +4,7 @@ module Metalhead using Flux, Images -export VGG19 +export load, VGG19 const imagenet_classes = split(String(read(joinpath(@__DIR__, "..", "imagenet_classes.txt"))), "\n", keep = false)