Spaces:

cuhgrel
/

nemo-tts-api

Sleeping

cuhgrel commited on Sep 29

Commit

e14982b

1 Parent(s): e77401a

update

Files changed (2) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def load_models():
                 })
                 models['bikol'] = FastPitchModel.restore_from(
-                    "models/fastpitch_bikol_repacked.nemo",
                     override_config_path=override_config,
                     strict=False
                 ).to(device)
@@ -80,7 +80,7 @@ def load_models():
                 try:
                     # Load model with map_location to avoid device issues
                     models['bikol'] = FastPitchModel.restore_from(
-                        "models/fastpitch_bikol_repacked.nemo",
                         map_location=device,
                         strict=False
                     )

                 })
                 models['bikol'] = FastPitchModel.restore_from(
+                    "models/fastpitch_bikol_fixed.nemo",
                     override_config_path=override_config,
                     strict=False
                 ).to(device)
                 try:
                     # Load model with map_location to avoid device issues
                     models['bikol'] = FastPitchModel.restore_from(
+                        "models/fastpitch_bikol_fixed.nemo",
                         map_location=device,
                         strict=False
                     )

requirements.txt CHANGED Viewed

@@ -31,4 +31,4 @@ webdataset
 matplotlib
 wrapt
 wget
-ruamel.yam

 matplotlib
 wrapt
 wget
+ruamel.yaml==0.17.32