Use input filter in image search setup. Input filter wasn't used earlier

This commit is contained in:
Debanjum Singh Solanky
2022-08-05 01:24:21 +03:00
parent ca5a8bd113
commit d5b43eb836
2 changed files with 16 additions and 9 deletions

View File

@@ -293,14 +293,14 @@ def run():
global config_file
config_file = args.config_file
# Store the verbose flag
global verbose
verbose = args.verbose
# Store the raw config data.
global config
config = args.config
# Store the verbose flag
global verbose
verbose = args.verbose
# Set device to GPU if available
device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")