Update app.py

app.py

@@ -1,9 +1,19 @@
 import gradio as gr
+from huggingface_hub import hf_hub_download
 import subprocess
 import tempfile
 import shutil
 
-def run_inference(config_path, ckpt_path, prompt_path):
+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):
+    repo_id = "hpcai-tech/Open-Sora"
+    
+    # Download the selected model
+    ckpt_path = download_model(repo_id, model_name)
+
     with open(config_path, 'r') as file:
         config_content = file.read()
     config_content = config_content.replace('prompt_path = "./assets/texts/t2v_samples.txt"', f'prompt_path = "{prompt_path}"')
@@ -30,9 +40,13 @@ def main():
     gr.Interface(
         fn=run_inference,
         inputs=[
-            gr.Textbox(label="Configuration Path"),
-            gr.Dropdown(choices=["./path/to/model1.ckpt", "./path/to/model2.ckpt", "./path/to/model3.ckpt"], label="Checkpoint Path"),
-            gr.Textbox(label="Prompt Path")
+            gr.Textbox(label="Configuration Path", default="configs/opensora/inference/16x256x256.py"),
+            gr.Dropdown(choices=[
+                "OpenSora-v1-16x256x256.pth",
+                "OpenSora-v1-HQ-16x256x256.pth",
+                "OpenSora-v1-HQ-16x512x512.pth"
+            ], label="Model Selection"),
+            gr.Textbox(label="Prompt Path", default="./assets/texts/t2v_samples.txt")
         ],
         outputs=[
             gr.Text(label="Status"),