Update app.py

app.py

@@ -6,6 +6,27 @@ import shutil
 import os
 import spaces
 
+from transformers import T5ForConditionalGeneration, T5Tokenizer
+import os
+
+def download_t5_model(model_id, save_directory):
+    # Modelin tokenizer'ını ve modeli indir
+    model = T5ForConditionalGeneration.from_pretrained(model_id)
+    tokenizer = T5Tokenizer.from_pretrained(model_id)
+
+    # Model ve tokenizer'ı belirtilen dizine kaydet
+    if not os.path.exists(save_directory):
+        os.makedirs(save_directory)
+    model.save_pretrained(save_directory)
+    tokenizer.save_pretrained(save_directory)
+
+# Model ID ve kaydedilecek dizin
+model_id = "DeepFloyd/t5-v1_1-xxl"
+save_directory = "pretrained_models/t5_ckpts/t5-v1_1-xxl"
+
+# Modeli indir
+download_t5_model(model_id, save_directory)
+
 def download_model(repo_id, model_name):
     model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
     return model_path