update whisper load

app.py

@@ -184,8 +184,17 @@ whole_model.load_state_dict(state_dict, strict=False)
 # Set the model to evaluation mode
 whole_model.eval()
 
-whisper_model = WhisperForConditionalGeneration.from_pretrained("whisper-small-local")
-whisper_processor = WhisperProcessor.from_pretrained("whisper-small-local")
+whisper_model_path = "whisper-small-local"
+if os.path.exists(whisper_model_path):
+    #print("Loading Whisper model from local storage...")
+    whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_path)
+    whisper_processor = WhisperProcessor.from_pretrained(whisper_model_path)
+else:
+    print("Downloading Whisper model from Hugging Face...")
+    whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
+    whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-small")
+    
+
 
 def transcribe_audio(audio):
     if isinstance(audio, str):  # If it's a file path