jisaacso219 commited on
Commit
7315b11
·
verified ·
1 Parent(s): fb5fb6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ model = AutoModelForCausalLM.from_pretrained(
19
  )
20
 
21
  # Load lightweight TTS
22
- tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", gpu=torch.cuda.is_available())
 
23
 
24
  def generate_shodan_response(user_input):
25
  system_prompt = (
 
19
  )
20
 
21
  # Load lightweight TTS
22
+ tts = TTS(model_name="tts_models/en/vctk/vits", gpu=False)
23
+
24
 
25
  def generate_shodan_response(user_input):
26
  system_prompt = (