Update modeling.py

modeling.py

@@ -3,6 +3,8 @@ import torchaudio
 import torch.nn as nn
 from transformers import PreTrainedModel
 import torch
+# download repo
+snapshot_download(repo_id="jadechoghari/VoiceRestore", repo_type="model")
 from BigVGAN import bigvgan
 from BigVGAN.meldataset import get_mel_spectrogram
 from voice_restore import VoiceRestore
@@ -31,8 +33,6 @@ class VoiceRestore(PreTrainedModel):
     
     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