Update app.py

app.py

@@ -42,10 +42,11 @@ def download_model(repo_id, model_name):
 import glob
 
 @spaces.GPU
-def run_inference(model_name, prompt_text):
+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",
@@ -110,14 +111,6 @@ def main():
 
         with gr.Row():
             with gr.Column():
-                model_dropdown = gr.Dropdown(
-                    choices=[
-                        "OpenSora-v1-16x256x256.pth",
-                        "OpenSora-v1-HQ-16x256x256.pth",
-                        "OpenSora-v1-HQ-16x512x512.pth"
-                    ], 
-                    value="OpenSora-v1-16x256x256.pth"
-                )
                 prompt_text = gr.Textbox(show_label=False, placeholder="Enter prompt text here", lines=4)
                 submit_button = gr.Button("Run Inference")
 
@@ -126,7 +119,7 @@ def main():
 
         submit_button.click(
             fn=run_inference, 
-            inputs=[model_dropdown, prompt_text], 
+            inputs=[prompt_text], 
             outputs=output_video
         )