From 760d234b613c4d5ea1b8c72ea9132f1c8bbbd8e9 Mon Sep 17 00:00:00 2001 From: Vladimir Date: Mon, 16 Dec 2024 15:50:04 +0100 Subject: [PATCH] Update scan.py --- scan.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/scan.py b/scan.py index 7b9292e..7aca1df 100644 --- a/scan.py +++ b/scan.py @@ -48,10 +48,6 @@ def scan_subdir(model_type): if filename.endswith('.pt'): available_models.append(filename) data_config_avai[filename] = scan_data_config(filename[:-3]) - elif model_type == 'tensorflow': - if filename.endswith('.pb'): - available_models.append(filename) - data_config_avai[filename] = scan_data_config(filename[:-3]) elif model_type == 'onnx': if filename.endswith('.onnx'): available_models.append(filename) @@ -62,7 +58,7 @@ def scan_subdir(model_type): def scan(): """Scan all directories that contain example models and write to a file.""" - available_types = ['keras', 'pytorch', 'onnx', 'tensorflow'] + available_types = ['keras', 'pytorch', 'onnx'] #Dictionary to save available model list model_dict = {} @@ -80,4 +76,4 @@ def scan(): json.dump(model_dict, open("available_models.json", 'w' )) json.dump(data_config_avai, open("available_data_config.json", 'w' )) -scan() \ No newline at end of file +scan()