TwentyNine commited on
Commit
4aedf7f
1 Parent(s): d153e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("TwentyNine/byt5-ain-kana-latin-co
6
 
7
  def transcribe(input_str):
8
  input_enc = tokenizer.encode(input_str, return_tensors='pt')
9
- output_enc = model.generate(input_ids, max_length=256)
10
  return tokenizer.decode(output_enc[0], skip_special_tokens=True)
11
 
12
  gradio_app = gr.Interface(
@@ -17,4 +17,4 @@ gradio_app = gr.Interface(
17
  )
18
 
19
  if __name__ == "__main__":
20
- gradio_app.launch()
 
6
 
7
  def transcribe(input_str):
8
  input_enc = tokenizer.encode(input_str, return_tensors='pt')
9
+ output_enc = model.generate(input_enc, max_length=256)
10
  return tokenizer.decode(output_enc[0], skip_special_tokens=True)
11
 
12
  gradio_app = gr.Interface(
 
17
  )
18
 
19
  if __name__ == "__main__":
20
+ gradio_app.launch(share=True)