Update app.py

app.py

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