Another003 commited on
Commit
8f90ee5
·
verified ·
1 Parent(s): f6ee992

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -3
main.py CHANGED
@@ -5,6 +5,7 @@ from googletrans import Translator
5
  from fastapi import FastAPI, File, UploadFile, HTTPException
6
  from fastapi.responses import JSONResponse
7
  from pathlib import Path
 
8
 
9
  app = FastAPI()
10
 
@@ -44,12 +45,16 @@ async def process_audio(file: UploadFile = File(...)):
44
  with open(file_location, "wb") as saved_file:
45
  saved_file.write(file.file.read())
46
 
 
 
 
 
47
  # JP
48
- original = pipe(file_location)
49
  original_version = original["text"]
50
 
51
  # EN
52
- result = pipe(file_location, generate_kwargs={"task": "translate"})
53
  hasil = result["text"]
54
 
55
  # ID
@@ -96,4 +101,4 @@ def modify_text(text):
96
  for original, replacement in replacements.items():
97
  text = text.replace(original, replacement)
98
 
99
- return text
 
5
  from fastapi import FastAPI, File, UploadFile, HTTPException
6
  from fastapi.responses import JSONResponse
7
  from pathlib import Path
8
+ import numpy as np
9
 
10
  app = FastAPI()
11
 
 
45
  with open(file_location, "wb") as saved_file:
46
  saved_file.write(file.file.read())
47
 
48
+ # Read audio file and convert to NumPy ndarray
49
+ audio_array, _ = torchaudio.load(file_location, normalize=True)
50
+ audio_array = np.array(audio_array[0].numpy())
51
+
52
  # JP
53
+ original = pipe(audio_array)
54
  original_version = original["text"]
55
 
56
  # EN
57
+ result = pipe(audio_array, generate_kwargs={"task": "translate"})
58
  hasil = result["text"]
59
 
60
  # ID
 
101
  for original, replacement in replacements.items():
102
  text = text.replace(original, replacement)
103
 
104
+ return text