Update app.py

app.py

@@ -26,11 +26,11 @@ fp16 = device != 'cpu'
 def transl(text, src='en', dest='hy' ):
     return translator.translate(text, src=src, dest=dest).text
     
-def generate(prompt: str):
+def generate(prompt: str, length=500):
             input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
             out = model.generate(input_ids, 
-                    min_length=300, 
-                    max_length=700, 
+                    min_length=length, 
+                    max_length=length, 
                     top_p=0.95, 
                     top_k=0,
                     temperature=0.9,
@@ -40,13 +40,14 @@ def generate(prompt: str):
             return generated_text #+ '\n\n'+transl(generated_text, src='hy', dest='en')
             
 length = gr.inputs.Slider(200, 1000)
+
 contexto = gr.inputs.Textbox(lines=10, placeholder="Write your text here")
 
 interface = gr.Interface.load("huggingface/sberbank-ai/mGPT-armenian",
             description=description,
             examples=examples, 
             fn=generate,
-            inputs=inputs=[contexto,length],#"text",
+            inputs=[contexto,length],#"text",
             outputs='text',
             thumbnail = 'https://habrastorage.org/r/w1560/getpro/habr/upload_files/26a/fa1/3e1/26afa13e1d1a56f54c7b0356761af7b8.png',
             theme = "peach",