Update app.py

app.py

@@ -10,14 +10,13 @@ from transformers import pipeline
 
 # functions
 def generate(promt, max_tokens):
-    if not description or not model:
+    if not promt:
         return ""
     
     language = detect(prompt)
     
-    if language == 'ru':
-        prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-        print(prompt)
+    prompt = GoogleTranslator(source=language, target='en').translate(prompt)
+    print(prompt)
 
     
     output = generator(prompt, max_tokens)[0]['generated_text']
@@ -44,7 +43,7 @@ with gr.Blocks(css=css) as vui:
                 with gr.Row():
                     with gr.Column(scale=3):
                         with gr.Row():
-                            max_tokens = gr.Slider(show_label=True, label="Максимальное количество токенов", minimum=100, maximum=15000, value=5000, step=1)
+                            max_tokens = gr.Slider(show_label=True, label="Длина ответа", minimum=10, maximum=2500, value=100, step=1)
             with gr.Column():
                 text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
             with gr.Column(scale=2):