Update download_models.py

download_models.py

@@ -0,0 +1,30 @@
+import whisper
+import os
+
+def download_and_save_whisper_model(model_name="tiny", save_directory="./whisper_models"):
+    """
+    Download a Whisper model and save it locally.
+    
+    :param model_name: Name of the Whisper model to download (default is "tiny")
+    :param save_directory: Directory to save the model (default is "./whisper_models")
+    :return: Path to the saved model
+    """
+    # Ensure the save directory exists
+    os.makedirs(save_directory, exist_ok=True)
+    
+    # Load the model (this will download it if not already present)
+    model = whisper.load_model(model_name)
+    
+    # Get the path where the model is saved
+    model_path = os.path.join(save_directory, f"whisper-{model_name}.pt")
+    
+    # Save the model
+    whisper.save_model(model, model_path)
+    
+    print(f"Model saved to: {model_path}")
+    return model_path
+
+# Example usage
+if __name__ == "__main__":
+    model_path = download_and_save_whisper_model("tiny")
+    print(f"You can now load the model from: {model_path}")