Update app.py

app.py

@@ -3,11 +3,14 @@ from huggingface_hub import hf_hub_download
 import subprocess
 import tempfile
 import shutil
+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_path):
     repo_id = "hpcai-tech/Open-Sora"