Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
 import subprocess
-import tempfile
+import tempfile, time
 import shutil
 import os
 import spaces
@@ -31,6 +31,18 @@ def download_model(repo_id, model_name):
 import glob
 
 @spaces.GPU(duration=500)
+def run_model(temp_config_path, ckpt_path):
+    start_time = time.time()  # Record the start time
+    cmd = [
+        "torchrun", "--standalone", "--nproc_per_node", "1",
+        "scripts/inference.py", temp_config_path,
+        "--ckpt-path", ckpt_path
+    ]
+    subprocess.run(cmd)
+    end_time = time.time()  # Record the end time
+    execution_time = end_time - start_time  # Calculate the execution time
+    print(f"Model Execution time: {execution_time} seconds")
+
 def run_inference(model_name, prompt_text):
     repo_id = "hpcai-tech/Open-Sora"
     
@@ -57,12 +69,7 @@ def run_inference(model_name, prompt_text):
         temp_file.write(config_content)
         temp_config_path = temp_file.name
 
-    cmd = [
-        "torchrun", "--standalone", "--nproc_per_node", "1",
-        "scripts/inference.py", temp_config_path,
-        "--ckpt-path", ckpt_path
-    ]
-    subprocess.run(cmd)
+    run_model(temp_config_path, ckpt_path)
 
     save_dir = "./outputs/samples/"  # Örneğin, inference.py tarafından kullanılan kayıt dizini
     list_of_files = glob.glob(f'{save_dir}/*')
@@ -77,6 +84,8 @@ def run_inference(model_name, prompt_text):
     os.remove(temp_file.name)
     os.remove(prompt_file.name)
 
+
+
 def main():
     gr.Interface(
         fn=run_inference,