ai-forever commited on
Commit
a9069b3
1 Parent(s): 84ff7f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def transl(text, src='en', dest='hy' ):
29
  def generate(prompt: str):
30
  #print(prompt)
31
  #print(length, type(length))
32
- length = int(length)
33
  input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
34
  out = model.generate(input_ids,
35
  min_length=500,
 
29
  def generate(prompt: str):
30
  #print(prompt)
31
  #print(length, type(length))
32
+ #length = int(length)
33
  input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
34
  out = model.generate(input_ids,
35
  min_length=500,