Spaces:
Runtime error
Runtime error
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:
|