Update gradio_app.py

gradio_app.py

@@ -9,6 +9,9 @@ is_shared_ui = True if "fffiloni/Paints-UNDO" in os.environ['SPACE_ID'] else Fal
 
 is_gpu_associated = torch.cuda.is_available()
 
+import subprocess
+from subprocess import getoutput
+
 if is_gpu_associated:
     gpu_info = getoutput('nvidia-smi')
     if("A10G" in gpu_info):
@@ -22,7 +25,6 @@ import functools
 import random
 import gradio as gr
 import numpy as np
-
 import wd14tagger
 import memory_management
 import uuid