Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,8 @@ import torchaudio
|
|
| 4 |
from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech
|
| 5 |
from transformers.models.speecht5 import SpeechT5HifiGan
|
| 6 |
|
|
|
|
|
|
|
| 7 |
# Load model and processor
|
| 8 |
processor = SpeechT5Processor.from_pretrained("nambn0321/TTS_british_2nd_attempt")
|
| 9 |
model = SpeechT5ForTextToSpeech.from_pretrained("nambn0321/TTS_british_2nd_attempt", use_safetensors=True, trust_remote_code=True)
|
|
|
|
| 4 |
from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech
|
| 5 |
from transformers.models.speecht5 import SpeechT5HifiGan
|
| 6 |
|
| 7 |
+
print(torch.cuda.is_available())
|
| 8 |
+
|
| 9 |
# Load model and processor
|
| 10 |
processor = SpeechT5Processor.from_pretrained("nambn0321/TTS_british_2nd_attempt")
|
| 11 |
model = SpeechT5ForTextToSpeech.from_pretrained("nambn0321/TTS_british_2nd_attempt", use_safetensors=True, trust_remote_code=True)
|