Update modeling.py

modeling.py

@@ -10,7 +10,6 @@ import argparse
 from model import OptimizedAudioRestorationModel
 import librosa
 from inference_long import apply_overlap_windowing_waveform, reconstruct_waveform_from_windows
-from safetensors.torch import load_file
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -46,7 +45,7 @@ class VoiceRestore(PreTrainedModel):
 
         # Optimized restoration model
         self.optimized_model = OptimizedAudioRestorationModel(device=device, bigvgan_model=self.bigvgan_model)
-        save_path = "./pytorch_model.safetensors"
+        save_path = "./model.safetensors"
         state_dict = load_file(save_path, device=device)
         if 'model_state_dict' in state_dict:
             state_dict = state_dict['model_state_dict']