fix Gradio demo error in "slow" visualization

Author: ntamotsu

app.py

@@ -1,3 +1,4 @@
+import imageio
 import gradio as gr
 import random
 import torch
@@ -18,6 +19,9 @@
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 import librosa
 
+os.environ['DISPLAY'] = ':0.0'
+os.environ['PYOPENGL_PLATFORM'] = 'egl'
+
 # Load model
 cfg = parse_args(phase="webui")  # parse config file
 cfg.FOLDER = 'cache'

requirements.txt

@@ -24,3 +24,4 @@ natsort
 pyrender
 moviepy
 librosa
+triangle==20220202