Update app.py

app.py

@@ -15,7 +15,8 @@ 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, max_length=200):
+    text += '\n'
     input_ids = tokenizer.encode(text, return_tensors="pt")
     with torch.no_grad():
         out = model.generate(input_ids,
@@ -32,8 +33,8 @@ 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_30epochs_1bs.bin')
 
 st.title("NeuroKorzh")
-# st.markdown("<img width=200px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",
-#             unsafe_allow_html=True)
+st.markdown("<img width=200px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",
+             unsafe_allow_html=True)
 
 st.markdown("\n")