Noumida commited on
Commit
ff42fba
·
verified ·
1 Parent(s): bafb16f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from __future__ import annotationsfrom __future__ import annotations
2
  import torch
3
  import torchaudio
4
  import gradio as gr
@@ -9,20 +9,19 @@ import re
9
  DESCRIPTION = "IndicConformer ASR with Automatic Language Identification"
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
- # --- ASR Model (The one we used before) ---
13
  print("Loading ASR model (IndicConformer)...")
14
  asr_model_id = "ai4bharat/indic-conformer-600m-multilingual"
15
  asr_model = AutoModel.from_pretrained(asr_model_id, trust_remote_code=True).to(device)
16
  asr_model.eval()
17
- print(" ASR Model loaded.")
18
 
19
- # --- Language Identification (LID) Model ---
20
  print("\nLoading Language ID model (MMS-LID)...")
21
  lid_model_id = "facebook/mms-lid"
22
  lid_processor = AutoProcessor.from_pretrained(lid_model_id)
23
  lid_model = AutoModel.from_pretrained(lid_model_id).to(device)
24
  lid_model.eval()
25
- print(" Language ID Model loaded.")
26
 
27
 
28
  # --- Language Mappings ---
@@ -53,7 +52,7 @@ print("Loading ASR model (IndicConformer)...")
53
  asr_model_id = "ai4bharat/indic-conformer-600m-multilingual"
54
  asr_model = AutoModel.from_pretrained(asr_model_id, trust_remote_code=True).to(device)
55
  asr_model.eval()
56
- print(" ASR Model loaded.")
57
 
58
  # --- Language Identification (LID) Model ---
59
  print("\nLoading Language ID model (MMS-LID)...")
@@ -61,7 +60,7 @@ lid_model_id = "facebook/mms-lid"
61
  lid_processor = AutoProcessor.from_pretrained(lid_model_id)
62
  lid_model = AutoModel.from_pretrained(lid_model_id).to(device)
63
  lid_model.eval()
64
- print(" Language ID Model loaded.")
65
 
66
 
67
  # --- Language Mappings ---
 
1
+ from __future__ import annotations
2
  import torch
3
  import torchaudio
4
  import gradio as gr
 
9
  DESCRIPTION = "IndicConformer ASR with Automatic Language Identification"
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
+
13
  print("Loading ASR model (IndicConformer)...")
14
  asr_model_id = "ai4bharat/indic-conformer-600m-multilingual"
15
  asr_model = AutoModel.from_pretrained(asr_model_id, trust_remote_code=True).to(device)
16
  asr_model.eval()
17
+ print(" ASR Model loaded.")
18
 
 
19
  print("\nLoading Language ID model (MMS-LID)...")
20
  lid_model_id = "facebook/mms-lid"
21
  lid_processor = AutoProcessor.from_pretrained(lid_model_id)
22
  lid_model = AutoModel.from_pretrained(lid_model_id).to(device)
23
  lid_model.eval()
24
+ print(" Language ID Model loaded.")
25
 
26
 
27
  # --- Language Mappings ---
 
52
  asr_model_id = "ai4bharat/indic-conformer-600m-multilingual"
53
  asr_model = AutoModel.from_pretrained(asr_model_id, trust_remote_code=True).to(device)
54
  asr_model.eval()
55
+ print(" ASR Model loaded.")
56
 
57
  # --- Language Identification (LID) Model ---
58
  print("\nLoading Language ID model (MMS-LID)...")
 
60
  lid_processor = AutoProcessor.from_pretrained(lid_model_id)
61
  lid_model = AutoModel.from_pretrained(lid_model_id).to(device)
62
  lid_model.eval()
63
+ print(" Language ID Model loaded.")
64
 
65
 
66
  # --- Language Mappings ---