Update modeling.py

modeling.py

@@ -10,6 +10,7 @@ import argparse
 from model import OptimizedAudioRestorationModel
 import librosa
 from inference_long import apply_overlap_windowing_waveform, reconstruct_waveform_from_windows
+from huggingface_hub import snapshot_download
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -27,9 +28,11 @@ class VoiceRestoreConfig(PretrainedConfig):
 # Model class for VoiceRestore
 class VoiceRestore(PreTrainedModel):
     config_class = VoiceRestoreConfig
-
+    
     def __init__(self, config: VoiceRestoreConfig):
         super().__init__(config)
+        # download repo
+        snapshot_download(repo_id="jadechoghari/VoiceRestore", repo_type="model")
         self.steps = config.steps
         self.cfg_strength = config.cfg_strength
         self.window_size_sec = config.window_size_sec