Update app.py

app.py

@@ -28,8 +28,8 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
                             num_beams=4,
                             temperature= temperature,
                             top_p=0.75,
-                            max_length=512,
-                            length_penalty = 5.5,
+                            max_length=356,
+                            length_penalty = 1.5,
                             eos_token_id = tokenizer.eos_token_id,
                             pad_token_id = tokenizer.pad_token_id,
                             num_return_sequences = 1,
@@ -38,7 +38,7 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
                             )
     decode = lambda x : tokenizer.decode(x, skip_special_tokens=True)
     generated_text = list(map(decode, out['sequences']))[0].replace(text,'')
-    return "Описание : "+generated_text
+    return generated_text
 
 def predict_t5(text, model, tokenizer, temperature=1.2):
     input_ids = tokenizer.encode(text, return_tensors="pt")
@@ -48,7 +48,7 @@ def predict_t5(text, model, tokenizer, temperature=1.2):
                         num_beams=4,
                         temperature=temperature,
                         top_p=0.35,
-                        max_length=512,
+                        max_length=356,
                         length_penalty = -1.0,
                         output_attentions = True,
                         return_dict_in_generate=True,