prithivMLmods commited on
Commit
0886910
·
verified ·
1 Parent(s): fca22b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ model_m = Qwen2VLForConditionalGeneration.from_pretrained(
61
  MODEL_ID,
62
  trust_remote_code=True,
63
  torch_dtype=torch.float16
64
- ).to("auto").eval()
65
 
66
  async def text_to_speech(text: str, voice: str, output_file="output.mp3"):
67
  """Convert text to speech using Edge TTS and save as MP3"""
 
61
  MODEL_ID,
62
  trust_remote_code=True,
63
  torch_dtype=torch.float16
64
+ ).to("cuda").eval()
65
 
66
  async def text_to_speech(text: str, voice: str, output_file="output.mp3"):
67
  """Convert text to speech using Edge TTS and save as MP3"""