Update app.py

app.py

@@ -8,9 +8,19 @@ def download_model(repo_id, model_name):
     model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
     return model_path
 
-def run_inference(config_path, model_name, prompt_path):
+def run_inference(model_name, prompt_path):
     repo_id = "hpcai-tech/Open-Sora"
     
+    # Map model names to their respective configuration files
+    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"
+    }
+    
+    # Get the configuration path based on the model name
+    config_path = config_mapping[model_name]
+    
     # Download the selected model
     ckpt_path = download_model(repo_id, model_name)
 
@@ -29,7 +39,7 @@ def run_inference(config_path, model_name, prompt_path):
     ]
     result = subprocess.run(cmd, capture_output=True, text=True)
 
-    shutil.rmtree(temp_config_path)
+    shutil.rmtree(temp_file.name)
 
     if result.returncode == 0:
         return "Inference completed successfully.", result.stdout
@@ -40,7 +50,6 @@ def main():
     gr.Interface(
         fn=run_inference,
         inputs=[
-            gr.Textbox(label="Configuration Path", value="configs/opensora/inference/16x256x256.py"),
             gr.Dropdown(choices=[
                 "OpenSora-v1-16x256x256.pth",
                 "OpenSora-v1-HQ-16x256x256.pth",