pere commited on
Commit
b22b7fe
1 Parent(s): 083d7ab

first submit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ def load_model():
12
  tokenizer = T5TokenizerFast.from_pretrained(model_name)
13
  return (model, tokenizer)
14
 
15
- def translate_to_nynorsk(model, tokenizer, text):
16
  encoded_txt = tokenizer(text, return_tensors="pt")
17
  generated_tokens = model.generate(
18
  **encoded_txt
@@ -34,6 +34,6 @@ st.text("Fixed text: ")
34
 
35
  if text:
36
  model, tokenizer = load_model()
37
- translated_text = translate_to_nynorsk(model, tokenizer, text)
38
  st.write(translated_text[0] if translated_text else "Unknown Error Translating Text")
39
 
 
12
  tokenizer = T5TokenizerFast.from_pretrained(model_name)
13
  return (model, tokenizer)
14
 
15
+ def deuncase(model, tokenizer, text):
16
  encoded_txt = tokenizer(text, return_tensors="pt")
17
  generated_tokens = model.generate(
18
  **encoded_txt
 
34
 
35
  if text:
36
  model, tokenizer = load_model()
37
+ translated_text = deuncase(model, tokenizer, text)
38
  st.write(translated_text[0] if translated_text else "Unknown Error Translating Text")
39