Qilex commited on
Commit
3420384
β€’
1 Parent(s): c9dfef5

Add multiline capabilities

Browse files
Files changed (1) hide show
  1. app.py +19 -4
app.py CHANGED
@@ -9,18 +9,33 @@ def translate(sentence):
9
  outputs = infer.generate(input_ids, max_new_tokens = len(sentence.split(' '))*10)
10
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
11
 
 
 
 
 
 
 
 
 
 
 
12
  title = "English to Middle English Translator"
13
  description = """
14
- This translator is trained on about 80,000 English/Middle English paired sentences.
15
  <br>
16
  It's still a work in progress.
 
17
  """
18
- article = ''''''
 
 
 
 
19
  gr.Interface(
20
- fn=translate,
21
  inputs=gr.Textbox(lines=1, placeholder="Enter text to translate."),
22
  outputs="text",
23
  title=title,
24
  description=description,
25
  article = article,
26
- ).launch()
 
9
  outputs = infer.generate(input_ids, max_new_tokens = len(sentence.split(' '))*10)
10
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
11
 
12
+ def translate_multiline(sentence):
13
+ if len(sentence.split() > 300):
14
+ print('Please insert less text')
15
+ if '\n' in sentence:
16
+ lines = sentence.split('\n')
17
+ translated_lines = [translate(line) for line in lines if len(line) > 0]
18
+ return '\n'.join(translated_lines)
19
+ else:
20
+ return translate(sentence)
21
+
22
  title = "English to Middle English Translator"
23
  description = """
24
+ This translator is trained on about 70,000 English/Middle English paired sentences.
25
  <br>
26
  It's still a work in progress.
27
+ <br>
28
  """
29
+ article = '''
30
+ <br>
31
+ You can improve results by removing conjunctions (hadn't -> had not)
32
+ '''
33
+
34
  gr.Interface(
35
+ fn=translate_multiline,
36
  inputs=gr.Textbox(lines=1, placeholder="Enter text to translate."),
37
  outputs="text",
38
  title=title,
39
  description=description,
40
  article = article,
41
+ ).launch()