ai-forever commited on
Commit
95590bf
1 Parent(s): a9069b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,8 +32,8 @@ def generate(prompt: str):
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,
36
- max_length=1000,
37
  top_p=0.95,
38
  top_k=0,
39
  temperature=0.9,
@@ -58,4 +58,4 @@ interface = gr.Interface.load("huggingface/sberbank-ai/mGPT-armenian",
58
  article = article
59
  )
60
 
61
- interface.launch(enable_queue=True)
 
32
  #length = int(length)
33
  input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
34
  out = model.generate(input_ids,
35
+ min_length=200,
36
+ max_length=700,
37
  top_p=0.95,
38
  top_k=0,
39
  temperature=0.9,
 
58
  article = article
59
  )
60
 
61
+ interface.launch(enable_queue=True)