Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,8 +32,8 @@ def translate(inputs, task):
|
|
| 32 |
raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
|
| 33 |
|
| 34 |
#text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
|
| 35 |
-
|
| 36 |
-
|
| 37 |
|
| 38 |
encoded_text = tokenizer(text, return_tensors="pt")
|
| 39 |
tokenizer.src_lang = "en_XX"
|
|
|
|
| 32 |
raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
|
| 33 |
|
| 34 |
#text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
|
| 35 |
+
ts_model = WhisperModel(MODEL_NAME, device = device, compute_type = "int8")
|
| 36 |
+
text = ts_model.transcribe(inputs, task = "translate")
|
| 37 |
|
| 38 |
encoded_text = tokenizer(text, return_tensors="pt")
|
| 39 |
tokenizer.src_lang = "en_XX"
|