model reulst print fix

app.py

@@ -37,8 +37,8 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
                             return_dict_in_generate=True,
                             )
     decode = lambda x : tokenizer.decode(x, skip_special_tokens=True)
-    generated_text = list(map(decode, out['sequences'])).split('Описание :')[1]
-    return generated_text
+    generated_text = list(map(decode, out['sequences']))[0].split('Описание :')[1]
+    return 'Описание : '+ generated_text
 
 def predict_t5(text, model, tokenizer, temperature=1.2):
     input_ids = tokenizer.encode(text, return_tensors="pt")
@@ -57,7 +57,7 @@ def predict_t5(text, model, tokenizer, temperature=1.2):
                         )
     decode = lambda x : tokenizer.decode(x, skip_special_tokens=True)
     generated_text = list(map(decode, out['sequences']))[0]
-    return generated_text
+    return 'Описание '+generated_text
 
 def generate(model,text):
     if model == 'GPT':
@@ -76,9 +76,11 @@ demo = gr.Interface(
     fn=generate,
     inputs=[
         gr.components.Dropdown(label="Модель", choices=('GPT', 'T5')),
-        gr.components.Textbox(label="Характеристики ноутбука",value = example,lines = example.count('\n')),
+        gr.components.Textbox(label="Характеристики",value = example,max_lines = example.count('\n')),
+    ],
+    outputs=[
+        gr.components.Textbox(label="Описание",interactive = False),
     ],
-    outputs=["text"],
 )
 
 demo.launch()