ercaronte commited on
Commit
65ecf8c
1 Parent(s): 2fbbc0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,7 +3,10 @@ import numpy as np
3
  import torch
4
  from datasets import load_dataset
5
 
6
- from transformers import pipeline, VitsModel, VitsTokenizer
 
 
 
7
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
 
3
  import torch
4
  from datasets import load_dataset
5
 
6
+ from transformers import pipeline
7
+ from transformers import VitsModel, VitsTokenizer
8
+ from transformers import SpeechT5Processor, SpeechT5Processor, SpeechT5HifiGan
9
+
10
 
11
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
12