Shanuka01 commited on
Commit
be8c738
·
1 Parent(s): ba7ce63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import VITSTokenizer, VITSForConditionalGeneration
5
 
6
  # Load the pre-trained VITS model and tokenizer
7
  model_name = "google/vit-base-patch16-224"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = VITSForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # Function to record a voice sample
 
5
 
6
  # Load the pre-trained VITS model and tokenizer
7
  model_name = "google/vit-base-patch16-224"
8
+ tokenizer = VITSTokenizer.from_pretrained(model_name)
9
  model = VITSForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # Function to record a voice sample