Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import tokenizers
 import streamlit as st
 import re
+from PIL import Image
 
 
 @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
@@ -32,18 +33,21 @@ 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=400px src='https://the-flow.ru/uploads/images/resize/830x0/adaptiveResize/05/06/06/42/25/8c7405840cd7.jpg'>",
-             unsafe_allow_html=True)
+# 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')
 
 st.markdown("\n")
 
-text = st.text_area(label='Starting point for text generation', height=100)
+text = st.text_input('Starting point for text generation', 'Что делать, Макс?', height=100)
 button = st.button('Go')
 
 if button:
     #try:
-    result = predict(text, model, tokenizer)
+    with st.spinner("Generation in progress"):
+        result = predict(text, model, tokenizer)
     
     #st.subheader('Max Korzh:')
     #lines = result.split('\n')
@@ -54,4 +58,4 @@ if button:
     st.write(lines)
     
     #except Exception:
-    #    st.error("Ooooops, something went wrong. Try again please and report to me, tg: @vladyur")
+    #    st.error("Ooooops, something went wrong. Try again please and report to me, tg: @vladyur")