TwentyNine
commited on
Commit
•
4aedf7f
1
Parent(s):
d153e3c
Update app.py
Browse files
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(
|
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)
|