Fiexd Model Path Error

app.py

@@ -2,23 +2,15 @@ import torch
 import torchaudio
 import gradio as gr
 import os
-import time
 import numpy as np
 
 from transformers import Wav2Vec2Processor, Wav2Vec2Model
 from safetensors.torch import load_file
 import torch.nn as nn
 
-# Wait for model.safetensors to be available
-model_path = "model.safetensors"
-timeout = 300  # wait up to 5 minutes
-start_time = time.time()
+from huggingface_hub import hf_hub_download
 
-while not os.path.exists(model_path):
-    if time.time() - start_time > timeout:
-        raise TimeoutError(f"{model_path} not found after {timeout} seconds.")
-    print(f"Waiting for {model_path} to be downloaded...")
-    time.sleep(5)
+model_path = hf_hub_download(repo_id="creativepurus/accent-wav2vec2", filename="model.safetensors")
 
 # Load processor
 processor = Wav2Vec2Processor.from_pretrained("creativepurus/accent-wav2vec2")

requirements.txt

@@ -2,4 +2,5 @@ torch
 transformers
 safetensors
 torchaudio
-gradio
+gradio
+huggingface_hub