Update app.py

app.py

@@ -20,7 +20,7 @@ def get_model(model_name, model_path):
 
 
 #@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
-def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_length=300):
+def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, length_of_generated=300):
     text += '\n'
     input_ids = tokenizer.encode(text, return_tensors="pt")
     length_of_prompt = len(input_ids[0])
@@ -30,7 +30,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_l
                              num_beams=n_beams,
                              temperature=temperature,
                              top_p=top_p,
-                             max_length=max_length + length_of_prompt,
+                             max_length=length_of_prompt + length_of_generated,
                              eos_token_id=tokenizer.eos_token_id
                              )
 
@@ -40,8 +40,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_l
 model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_best_eval_loss.bin')
 
 # st.title("NeuroKorzh")
-# st.markdown("<img width=400px src='https://the-flow.ru/uploads/images/resize/830x0/adaptiveResize/05/06/06/42/25/8c7405840cd7.jpg'>",
-#              unsafe_allow_html=True)
+
 image = Image.open('korzh.jpg')
 st.image(image, caption='NeuroKorzh')
 
@@ -63,7 +62,7 @@ if button:
     #lines = result.replace('\n', '\n\n')
     #st.write(lines)
     
-    st.text_area(label='', value=result, height=400)
+    st.text_area(label='', value=result, height=1200)
     
     #except Exception:
     #    st.error("Ooooops, something went wrong. Try again please and report to me, tg: @vladyur")