Update app.py

app.py

@@ -11,17 +11,13 @@ def download_model(repo_id, model_name):
 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)
 
     with open(config_path, 'r') as file:
@@ -37,14 +33,12 @@ def run_inference(model_name, prompt_path):
         "scripts/inference.py", temp_config_path,
         "--ckpt-path", ckpt_path
     ]
-    result = subprocess.run(cmd, capture_output=True, text=True)
-
+    subprocess.run(cmd, capture_output=True, text=True)
     shutil.rmtree(temp_file.name)
 
-    if result.returncode == 0:
-        return "Inference completed successfully.", result.stdout
-    else:
-        return "Error occurred:", result.stderr
+    # Assuming the output video is saved in a known path, e.g., "./output/video.mp4"
+    output_video_path = "./output/video.mp4"
+    return output_video_path
 
 def main():
     gr.Interface(
@@ -57,10 +51,7 @@ def main():
             ], label="Model Selection"),
             gr.Textbox(label="Prompt Path", value="./assets/texts/t2v_samples.txt")
         ],
-        outputs=[
-            gr.Text(label="Status"),
-            gr.Text(label="Output")
-        ],
+        outputs=gr.Video(label="Output Video"),
         title="Open-Sora Inference",
         description="Run Open-Sora Inference with Custom Parameters"
     ).launch()