Update app.py

app.py

@@ -2,6 +2,7 @@ import transformers
 import torch
 import tokenizers
 import streamlit as st
+import re
 
 
 @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, _regex.Pattern: lambda _: None}, suppress_st_warning=True)
@@ -31,8 +32,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")