Fix: use MultiTaskSpeakerModel and remove theme parameter for Gradio compatibility
Browse files
app.py
CHANGED
|
@@ -71,10 +71,10 @@ class MultiModelProfiler:
|
|
| 71 |
from transformers import Wav2Vec2FeatureExtractor
|
| 72 |
processor = Wav2Vec2FeatureExtractor.from_pretrained(encoder_name)
|
| 73 |
|
| 74 |
-
# Load model
|
| 75 |
-
from src.models import
|
| 76 |
|
| 77 |
-
model =
|
| 78 |
model_name=encoder_name,
|
| 79 |
num_gender_classes=2,
|
| 80 |
num_dialect_classes=3,
|
|
@@ -218,7 +218,8 @@ def create_interface():
|
|
| 218 |
status = "β" if name in profiler.models else "β"
|
| 219 |
model_info += f"{status} **{name}**: {config['description']}\n"
|
| 220 |
|
| 221 |
-
|
|
|
|
| 222 |
gr.Markdown(
|
| 223 |
"""
|
| 224 |
# ποΈ Vietnamese Speaker Profiling
|
|
|
|
| 71 |
from transformers import Wav2Vec2FeatureExtractor
|
| 72 |
processor = Wav2Vec2FeatureExtractor.from_pretrained(encoder_name)
|
| 73 |
|
| 74 |
+
# Load model - use MultiTaskSpeakerModel
|
| 75 |
+
from src.models import MultiTaskSpeakerModel
|
| 76 |
|
| 77 |
+
model = MultiTaskSpeakerModel(
|
| 78 |
model_name=encoder_name,
|
| 79 |
num_gender_classes=2,
|
| 80 |
num_dialect_classes=3,
|
|
|
|
| 218 |
status = "β" if name in profiler.models else "β"
|
| 219 |
model_info += f"{status} **{name}**: {config['description']}\n"
|
| 220 |
|
| 221 |
+
# Use gr.Blocks without theme for compatibility with older Gradio
|
| 222 |
+
with gr.Blocks(title="Vietnamese Speaker Profiling") as demo:
|
| 223 |
gr.Markdown(
|
| 224 |
"""
|
| 225 |
# ποΈ Vietnamese Speaker Profiling
|