Update app.py

app.py

@@ -78,14 +78,44 @@ Please consider citing the following papers if you find our work helpful.
 
 device = "cuda"
 
-# device = "cuda" if torch.cuda.is_available() else "cpu"
+try:
+    s2l = Speech2Language.from_pretrained(
+        model_tag=f"espnet/owsm_v4_medium_1B",
+        device="cpu",
+        nbest=1,
+    )
+except Exception as e:
+    print("File downloaded")
 
+# 2. Remove unrequired file
+import yaml
+from pathlib import Path
+import espnet_model_zoo
+
+d = "models--espnet--owsm_v4_medium_1B/snapshots/471418ddaf0b03c9ab1fd75f1f5d26fc3aea3aa9/exp/s2t_train_conv2d8_size1024_e18_d18_mel128_raw_bpe50000/config.yaml"
+p = Path(espnet_model_zoo.__file__)
+config_path = p.parent / d
+
+def remove_key(obj, key="gradient_checkpoint_layers"):
+    if isinstance(obj, dict):
+        if key in obj:
+            del obj[key]
+        for k, v in list(obj.items()):
+            remove_key(v, key)
+    elif isinstance(obj, list):
+        for item in obj:
+            remove_key(item, key)
+
+with open(config_path, "r") as f:
+    config = yaml.safe_load(f)
+
+remove_key(config)
+
+with open(config_path, "w") as f:
+    yaml.safe_dump(config, f, sort_keys=False, allow_unicode=True)
+
+print("Done! All 'gradient_checkpoint_layers' keys removed.")
 
-s2l = Speech2Language.from_pretrained(
-    model_tag=f"espnet/owsm_v4_medium_1B",
-    device=device,
-    nbest=1,
-)
 
 s2t_ar = ARSpeech2Text.from_pretrained(
     model_tag=f"espnet/owsm_v4_medium_1B",