Update app.py

app.py

@@ -5,7 +5,6 @@ import streamlit as st
 import re
 
 from PIL import Image
-from huggingface_hub import hf_hub_download
 
 
 @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
@@ -21,7 +20,6 @@ def get_model(model_name, model_path):
     return model, tokenizer
 
 
-#@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, length_of_generated=300):
     text += '\n'
     input_ids = tokenizer.encode(text, return_tensors="pt")
@@ -45,34 +43,27 @@ large_model, large_tokenizer = get_model('sberbank-ai/rugpt3large_based_on_gpt2'
 # st.title("NeuroKorzh")
 
 image = Image.open('korzh.jpg')
-st.image(image, caption='NeuroKorzh')
+st.image(image, caption='НейроКорж')
 
-option = st.selectbox('Model to be used', ('medium', 'large'))
+option = st.selectbox('Выберите своего Коржа', ('Быстрый', 'Глубокий'))
+craziness = st.slider(label='Абсурдность', min_value=0.5, max_value=4, value=2.5, step=0.1)
 
 st.markdown("\n")
 
-text = st.text_area(label='Starting point for text generation', value='Что делать, Макс?', height=200)
-button = st.button('Go')
+text = st.text_area(label='Напишите начало песни', value='Что делать, Макс?', height=100)
+button = st.button('Старт')
 
 if button:
-    #try:
-    with st.spinner("Generation in progress"):
-        if option == 'medium':
-            result = predict(text, medium_model, medium_tokenizer)
-        elif option == 'large':
-            result = predict(text, large_model, large_tokenizer)
-        else:
-            raise st.error('Error in selectbox')
+    try:
+        with st.spinner("Пушечка пишется"):
+            if option == 'Быстрый':
+                result = predict(text, medium_model, medium_tokenizer, temperature=craziness)
+            elif option == 'Глубокий':
+                result = predict(text, large_model, large_tokenizer, temperature=craziness)
+            else:
+                raise st.error('Error in selectbox')
+        
+        st.text_area(label='', value=result, height=1100)
     
-    #st.subheader('Max Korzh:')
-    #lines = result.split('\n')
-    #for line in lines:
-    #    st.write(line)
-    
-    #lines = result.replace('\n', '\n\n')
-    #st.write(lines)
-    
-    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")
+    except Exception:
+        st.error("Ooooops, something went wrong. Please try again and report to me, tg: @vladyur")