Update app.py

app.py

@@ -29,21 +29,21 @@ def download_model(repo_id, model_name):
 import glob
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+repo_id = "hpcai-tech/Open-Sora"
+
+# Map model names to their respective configuration files
+model_name = "OpenSora-v1-HQ-16x512x512.pth"
+config_mapping = {
+    "OpenSora-v1-16x256x256.pth": "configs/opensora/inference/16x256x256.py",
+    "OpenSora-v1-HQ-16x256x256.pth": "configs/opensora/inference/16x512x512.py",
+    "OpenSora-v1-HQ-16x512x512.pth": "configs/opensora/inference/64x512x512.py"
+}
+
+config_path = config_mapping[model_name]
+ckpt_path = download_model(repo_id, model_name)
 
 @spaces.GPU(duration=200)
 def run_inference(prompt_text):
-    repo_id = "hpcai-tech/Open-Sora"
-    
-    # Map model names to their respective configuration files
-    model_name = "OpenSora-v1-HQ-16x512x512.pth"
-    config_mapping = {
-        "OpenSora-v1-16x256x256.pth": "configs/opensora/inference/16x256x256.py",
-        "OpenSora-v1-HQ-16x256x256.pth": "configs/opensora/inference/16x512x512.py",
-        "OpenSora-v1-HQ-16x512x512.pth": "configs/opensora/inference/64x512x512.py"
-    }
-    
-    config_path = config_mapping[model_name]
-    ckpt_path = download_model(repo_id, model_name)
 
     # Save prompt_text to a temporary text file
     prompt_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w')