Create download_model.py

Created download_model.py

download_model.py

@@ -0,0 +1,16 @@
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+import os
+
+model_name = "facebook/blenderbot-400M-distill"
+model_dir = "/home/user/app/blenderbot_model"
+
+if not os.path.exists(model_dir):
+    print(f"Downloading {model_name}...")
+    tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
+    model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
+    os.makedirs(model_dir, exist_ok=True)
+    tokenizer.save_pretrained(model_dir)
+    model.save_pretrained(model_dir)
+    print("Download complete.")
+else:
+    print(f"Model already exists at {model_dir}.")