Update app.py

app.py

@@ -1,11 +1,46 @@
 import os
+import subprocess
 import logging
 import re
-import spaces
 import random
 import string
-import torch
 import requests
+import sys
+import warnings
+
+warnings.filterwarnings("ignore")
+root = logging.getLogger()
+root.setLevel(logging.DEBUG)
+handler = logging.StreamHandler(sys.stdout)
+handler.setLevel(logging.DEBUG)
+formatter = logging.Formatter('\n >>> [%(levelname)s] %(asctime)s %(name)s: %(message)s\n')
+handler.setFormatter(formatter)
+root.addHandler(handler)
+handler2 = logging.StreamHandler(sys.stderr)
+handler2.setLevel(logging.DEBUG)
+formatter = logging.Formatter('\n >>> [%(levelname)s] %(asctime)s %(name)s: %(message)s\n')
+handler2.setFormatter(formatter)
+root.addHandler(handler2)
+
+def cmd(cmd, assert_success=False, capture_output=False, env=None, dry_run=False):
+    if dry_run:
+        print(f"--> {cmd}")
+        result = 1
+    else:
+        result = subprocess.run(cmd, shell=True, capture_output=capture_output, env=env)
+        if assert_success and result.returncode != 0:
+            logging.error(
+                f"Command '{cmd}' failed with exit status code '{result.returncode}'. Exiting..."
+            )
+            sys.exit()
+
+    return result
+
+cmd("sudo apt install python3-mpi4py")
+cmd("")
+
+import spaces
+import torch
 import gradio as gr
 import numpy as np
 from lxml.html import fromstring
@@ -23,23 +58,6 @@ from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, Eul
 #import jax
 #import jax.numpy as jnp
 from pyina.launchers import TorqueMpiPool as Pool
-import sys
-import warnings
-
-warnings.filterwarnings("ignore")
-
-root = logging.getLogger()
-root.setLevel(logging.DEBUG)
-handler = logging.StreamHandler(sys.stdout)
-handler.setLevel(logging.DEBUG)
-formatter = logging.Formatter('\n >>> [%(levelname)s] %(asctime)s %(name)s: %(message)s\n')
-handler.setFormatter(formatter)
-root.addHandler(handler)
-handler2 = logging.StreamHandler(sys.stderr)
-handler2.setLevel(logging.DEBUG)
-formatter = logging.Formatter('\n >>> [%(levelname)s] %(asctime)s %(name)s: %(message)s\n')
-handler2.setFormatter(formatter)
-root.addHandler(handler2)
 
 last_motion=None
 dtype = torch.float16