Update modeling.py

modeling.py

@@ -3,10 +3,6 @@ import torchaudio
 import torch.nn as nn
 from transformers import PreTrainedModel
 import torch
-# download repo
-from huggingface_hub import snapshot_download
-repo_dir = snapshot_download(repo_id="jadechoghari/VoiceRestore", repo_type="model", local_dir="VoiceRestoreRepo")
-sys.path.append(repo_dir)
 from BigVGAN import bigvgan
 from BigVGAN.meldataset import get_mel_spectrogram
 from voice_restore import VoiceRestore
@@ -52,10 +48,8 @@ class VoiceRestore(PreTrainedModel):
         self.optimized_model = OptimizedAudioRestorationModel(device=device, bigvgan_model=self.bigvgan_model)
         save_path = "/content/voicerestore/checkpoints/voice-restore-20d-16h-optim.pt"
         state_dict = torch.load(save_path, map_location=torch.device(device))
-        print("loaded")
         if 'model_state_dict' in state_dict:
             state_dict = state_dict['model_state_dict']
-            print("change keys")
         
         self.optimized_model.voice_restore.load_state_dict(state_dict, strict=True)
         self.optimized_model.eval()