Update app.py

app.py

@@ -4,11 +4,14 @@ import subprocess
 import tempfile
 import shutil
 import os
+import spaces
 
 def download_model(repo_id, model_name):
     model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
     return model_path
 
+
+@spaces.GPU
 def run_inference(model_name, prompt_text):
     repo_id = "hpcai-tech/Open-Sora"
     
@@ -32,7 +35,6 @@ def run_inference(model_name, prompt_text):
         config_content = file.read()
     config_content = config_content.replace('prompt_path = "./assets/texts/t2v_samples.txt"', f'prompt_path = "{prompt_file.name}"')
     
-    # Create a temporary file for the updated configuration
     with tempfile.NamedTemporaryFile('w', delete=False) as temp_file:
         temp_file.write(config_content)
         temp_config_path = temp_file.name
@@ -49,9 +51,12 @@ def run_inference(model_name, prompt_text):
     os.remove(prompt_file.name)
 
     if result.returncode == 0:
-        return "Inference completed successfully.", result.stdout
+        # Assuming the output video is saved at a known location, for example "./output/video.mp4"
+        output_video_path = "./output/video.mp4"
+        return output_video_path
     else:
-        return "Error occurred:", result.stderr
+        print("Error occurred:", result.stderr)
+        return None  # You might want to handle errors differently
 
 def main():
     gr.Interface(
@@ -64,7 +69,7 @@ def main():
             ], label="Model Selection"),
             gr.Textbox(label="Prompt Text", placeholder="Enter prompt text here")
         ],
-        outputs="text",
+        outputs=gr.Video(label="Output Video"),
         title="Open-Sora Inference",
         description="Run Open-Sora Inference with Custom Parameters",
     ).launch()