nambn0321 commited on
Commit
f2dea3d
·
verified ·
1 Parent(s): 1add378

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)