sanjitaa commited on
Commit
679e261
·
1 Parent(s): b0a4863

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- 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"
 
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"