Update app.py

app.py

@@ -7,9 +7,23 @@ from pydub import AudioSegment
 from pydub.silence import detect_nonsilent
 from waitress import serve
 from simple_salesforce import Salesforce
+import time
+from transformers import pipeline
 
 app = Flask(__name__)
 
+retry_attempts = 3
+timeout = 60  # 1 minute timeout for each attempt
+model = None
+for attempt in range(retry_attempts):
+    try:
+        model = pipeline("automatic-speech-recognition", model="openai/whisper-small", device=0 if torch.cuda.is_available() else -1, config={"timeout": timeout})
+        print("Model loaded successfully!")
+        break
+    except requests.exceptions.ReadTimeout:
+        print(f"Timeout occurred, retrying attempt {attempt + 1}/{retry_attempts}...")
+        time.sleep(5)  # Retry after 5 seconds
+
 # Use whisper-small for faster processing and better speed
 device = "cuda" if torch.cuda.is_available() else "cpu"
 asr_model = pipeline("automatic-speech-recognition", model="openai/whisper-small", device=0 if device == "cuda" else -1)