Thanh-Lam commited on
Commit
3c8c19b
Β·
1 Parent(s): f9308ba

Fix: use MultiTaskSpeakerModel and remove theme parameter for Gradio compatibility

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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 SpeakerProfileModel
76
 
77
- model = SpeakerProfileModel(
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
- with gr.Blocks(title="Vietnamese Speaker Profiling", theme=gr.themes.Soft()) as demo:
 
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