VictorKai1996NUS commited on
Commit
d6e8791
·
verified ·
1 Parent(s): b71d548

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,6 +11,8 @@ import logging
11
  import gradio as gr
12
  from videosys import CogVideoConfig, VideoSysEngine
13
  from videosys.models.cogvideo.pipeline import CogVideoPABConfig
 
 
14
 
15
  logging.basicConfig(level=logging.INFO)
16
  logger = logging.getLogger(__name__)
 
11
  import gradio as gr
12
  from videosys import CogVideoConfig, VideoSysEngine
13
  from videosys.models.cogvideo.pipeline import CogVideoPABConfig
14
+ import psutil
15
+ import GPUtil
16
 
17
  logging.basicConfig(level=logging.INFO)
18
  logger = logging.getLogger(__name__)