milyiyo commited on
Commit
8ca86e2
1 Parent(s): 87938f5

Remove the lang parameter

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ tokenizer = AutoTokenizer.from_pretrained("milyiyo/paraphraser-german-mt5-small"
11
  model = AutoModelForSeq2SeqLM.from_pretrained("milyiyo/paraphraser-german-mt5-small")
12
 
13
 
14
- def paraphrase(sentence: str, lang: str, count: str):
15
  p_count = int(count)
16
  if p_count <= 0 or len(sentence.strip()) == 0:
17
  return {'result': []}
@@ -35,7 +35,7 @@ def paraphrase(sentence: str, lang: str, count: str):
35
  res.append(line)
36
  return {'result': res}
37
 
38
- def paraphrase_dummy(sentence: str, lang: str, count: str):
39
  return {'result': []}
40
 
41
 
 
11
  model = AutoModelForSeq2SeqLM.from_pretrained("milyiyo/paraphraser-german-mt5-small")
12
 
13
 
14
+ def paraphrase(sentence: str, count: str):
15
  p_count = int(count)
16
  if p_count <= 0 or len(sentence.strip()) == 0:
17
  return {'result': []}
 
35
  res.append(line)
36
  return {'result': res}
37
 
38
+ def paraphrase_dummy(sentence: str, count: str):
39
  return {'result': []}
40
 
41