Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,8 @@ from transformers import pipeline
|
|
| 4 |
from pytube import YouTube
|
| 5 |
from datasets import Dataset, Audio
|
| 6 |
from moviepy.editor import AudioFileClip
|
|
|
|
|
|
|
| 7 |
|
| 8 |
pipe = pipeline(model="Neprox/model")
|
| 9 |
|
|
@@ -70,7 +72,10 @@ def get_translation(text):
|
|
| 70 |
Translates the given Swedish text to English.
|
| 71 |
"""
|
| 72 |
# TODO: Make API call to Google Translate to get English translation
|
| 73 |
-
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
def transcribe(audio, url, seconds_max):
|
| 76 |
"""
|
|
|
|
| 4 |
from pytube import YouTube
|
| 5 |
from datasets import Dataset, Audio
|
| 6 |
from moviepy.editor import AudioFileClip
|
| 7 |
+
import googletrans
|
| 8 |
+
from googletrans import Translator
|
| 9 |
|
| 10 |
pipe = pipeline(model="Neprox/model")
|
| 11 |
|
|
|
|
| 72 |
Translates the given Swedish text to English.
|
| 73 |
"""
|
| 74 |
# TODO: Make API call to Google Translate to get English translation
|
| 75 |
+
translator = Translator()
|
| 76 |
+
result = translator.translate(text)
|
| 77 |
+
|
| 78 |
+
return result
|
| 79 |
|
| 80 |
def transcribe(audio, url, seconds_max):
|
| 81 |
"""
|