kumararvindibs commited on
Commit
9dbfb6e
·
verified ·
1 Parent(s): 86fb543

Delete handlerForAudio.py

Browse files
Files changed (1) hide show
  1. handlerForAudio.py +0 -40
handlerForAudio.py DELETED
@@ -1,40 +0,0 @@
1
- from typing import Dict, Any
2
- from textToStoryGeneration import *
3
- import logging
4
- import torch
5
- import soundfile as sf
6
- from transformers import AutoTokenizer, AutoModelForTextToWaveform
7
-
8
- # Configure logging
9
- logging.basicConfig(level=logging.DEBUG)
10
- # Configure logging
11
- logging.basicConfig(level=logging.ERROR)
12
- # Configure logging
13
- logging.basicConfig(level=logging.WARNING)
14
-
15
-
16
-
17
- class CustomHandler:
18
- def __init__(self):
19
-
20
- self.tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
21
- self.model= AutoModelForTextToWaveform.from_pretrained("facebook/mms-tts-eng")
22
- def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
23
- # Prepare the payload with input data
24
- logging.warning(f"------input_data-- {str(data)}")
25
- payload = str(data)
26
- logging.warning(f"payload----{str(payload)}")
27
- # Set headers with API token
28
- inputs = self.tokenizer(payload, return_tensors="pt")
29
-
30
- # Generate the waveform from the input text
31
- with torch.no_grad():
32
- outputs = self.model(**inputs)
33
-
34
- # Save the audio to a file
35
- sf.write("StoryAudio.wav", outputs["waveform"][0].numpy(), self.model.config.sampling_rate)
36
-
37
- return 'StoryAudio.wav'
38
- # Check if the request was successful
39
-
40
-