Upload folder using huggingface_hub

README.md

@@ -1,2 +1 @@
-VERA
-
+VERA

app/app.py

@@ -1,43 +1,3 @@
-Hugging Face's logo
-Hugging Face
-Models
-Datasets
-Spaces
-Community
-Docs
-Enterprise
-Pricing
-
-
-master-of-hardcore
-/
-211048k-dnpvr 
-
-like
-0
-Model card
-Files and versions
-xet
-Community
-211048k-dnpvr
-/
-app
-/
-app.py
-
-QuangMing
-...
-2ca7e4f
-5 days ago
-raw
-
-Copy download link
-history
-blame
-contribute
-delete
-
-10.3 kB
 import fastapi
 import shutil
 import os
@@ -71,7 +31,16 @@ class ModelAPI:
         self.noisy_audio_path = os.path.join(self.base_path, "noisy_audio")
         self.enhanced_audio_path = os.path.join(self.base_path, "enhanced_audio")
         app_dir = os.path.dirname(os.path.abspath(__file__))
-        self.ckpt_path = glob.glob(os.path.join(app_dir, "*.ckpt"))[0]
+        
+        ckpt_files = glob.glob(os.path.join(app_dir, "*.ckpt"))
+
+        if not ckpt_files:
+            raise FileNotFoundError("No .ckpt file found in app_dir.")
+        elif len(ckpt_files) > 1:
+            raise RuntimeError("Multiple .ckpt files found in app_dir. Please keep only one.")
+        else:
+            self.ckpt_path = ckpt_files[0]
+        
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.corrector = "ald"
         self.corrector_steps = 1
@@ -322,4 +291,3 @@ class ModelAPI:
     def run(self):
 
         uvicorn.run(self.app, host=self.host, port=self.port)
-

app/derev_elevenlabs_7.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ddc75f01c3bc7d94d77333bf8ad93403c4bd0362279fc003177b3d50ddefa7a
-size 1295842026
+oid sha256:f6f18452b4a5a2a6aff0c2db0c236cc7804ca9ce2de5d8fb794b909e4f3993f5
+size 1295832473