ggirishg commited on
Commit
ce8c430
·
verified ·
1 Parent(s): ec512e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -5,13 +5,26 @@ from transformers import Wav2Vec2FeatureExtractor, Wav2Vec2Model
5
  import torchaudio
6
  from torchaudio.transforms import Resample
7
  from tensorflow.keras.models import load_model
8
-
9
  # Load models
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/mms-1b")
12
  wav2vec_model = Wav2Vec2Model.from_pretrained("facebook/mms-1b").to(device)
13
 
14
- saved_model_path = "CNN-MODEL"
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  try:
16
  cnn_model = load_model(saved_model_path)
17
  except Exception as e:
 
5
  import torchaudio
6
  from torchaudio.transforms import Resample
7
  from tensorflow.keras.models import load_model
8
+ from tensorflow.keras.models import load_model
9
  # Load models
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/mms-1b")
12
  wav2vec_model = Wav2Vec2Model.from_pretrained("facebook/mms-1b").to(device)
13
 
14
+
15
+
16
+
17
+ # h5_model_path = "model.h5"
18
+
19
+ # try:
20
+ # loaded_h5_model = load_model(h5_model_path)
21
+ # print("Model loaded successfully from .h5 format.")
22
+ # except Exception as e:
23
+ # print(f"Error: {e}")
24
+
25
+
26
+
27
+ saved_model_path = "model.h5"
28
  try:
29
  cnn_model = load_model(saved_model_path)
30
  except Exception as e: