Hugging Face's logo

Spaces:
mpc001
/
auto_avsr
Runtime error

App Files Community
1
mpc001 commited on
Commit
d577e1e
·
1 Parent(s): 4c034e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,9 +10,9 @@ import gradio as gr
10
  from pipelines.pipeline import InferencePipeline
11
 
12
  pipelines = {
13
- "VSR(mediapipe)": InferencePipeline("./configs/LRS3_V_WER19.1.ini", device="cuda:0", face_track=True, detector="mediapipe"),
14
- "ASR": InferencePipeline("./configs/LRS3_A_WER1.0.ini", device="cuda:0", face_track=True, detector="retinaface"),
15
- "AVSR(mediapipe)": InferencePipeline("./configs/LRS3_AV_WER0.9.ini", device="cuda:0", face_track=True, detector="mediapipe")
16
  }
17
  print("Step 0. Model has been loaded.")
18
 
 
10
  from pipelines.pipeline import InferencePipeline
11
 
12
  pipelines = {
13
+ "VSR(mediapipe)": InferencePipeline("./configs/LRS3_V_WER19.1.ini", device="cpu", face_track=True, detector="mediapipe"),
14
+ "ASR": InferencePipeline("./configs/LRS3_A_WER1.0.ini", device="cpu", face_track=True, detector="mediapipe"),
15
+ "AVSR(mediapipe)": InferencePipeline("./configs/LRS3_AV_WER0.9.ini", device="cpu", face_track=True, detector="mediapipe")
16
  }
17
  print("Step 0. Model has been loaded.")
18