Update app.py

app.py

@@ -22,7 +22,9 @@ model_id = "DeepFloyd/t5-v1_1-xxl"
 save_directory = "pretrained_models/t5_ckpts/t5-v1_1-xxl"
 
 # Modeli indir
+st_time_t5 = time.time()
 download_t5_model(model_id, save_directory)
+print(f"T5 Download Time : {st_time_t5-time.time()} seconds")
 
 def download_model(repo_id, model_name):
     model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
@@ -54,8 +56,10 @@ def run_inference(model_name, prompt_text):
     }
     
     config_path = config_mapping[model_name]
+    st_time_sora = time.time()
     ckpt_path = download_model(repo_id, model_name)
-
+    print(f"Open-Sora Download Time : {st_time_sora-time.time()} seconds")
+    
     # Save prompt_text to a temporary text file
     prompt_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w')
     prompt_file.write(prompt_text)