Update README.md

README.md

@@ -54,12 +54,10 @@ def mean_pooling(model_output, attention_mask):
     return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
 
 def download_onnx_model(model_name, cache_folder, model_path, force_download = False):
-    if force_download:
-       if os.path.exists(model_path):
-          shutil.rmtree(model_path)
-    else:
-       if os.path.exists(model_path):
-          return
+    if force_download and os.path.exists(model_path):
+       shutil.rmtree(model_path)
+    elif os.path.exists(model_path):
+       return
         
     snapshot_download(model_name,
                      cache_dir=cache_folder,