Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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:
|