Update gradio_app.py

gradio_app.py

@@ -7,7 +7,7 @@ os.makedirs(result_dir, exist_ok=True)
 
 is_shared_ui = True if "fffiloni/Paints-UNDO" in os.environ['SPACE_ID'] else False
 
-is_gpu_associated = torch.cuda.is_available()
+is_gpu_associated = torch.cuda.is_available() else False
 
 import subprocess
 from subprocess import getoutput
@@ -26,7 +26,10 @@ import random
 import gradio as gr
 import numpy as np
 import wd14tagger
-import memory_management
+
+if is_gpu_associated:
+    import memory_management
+
 import uuid
 
 from PIL import Image