Update app.py

app.py

@@ -33,8 +33,9 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, lengt
                              max_length=length_of_prompt + length_of_generated,
                              eos_token_id=tokenizer.eos_token_id
                              )
-
-    return list(map(tokenizer.decode, out))[0]
+                          
+    generated = list(map(tokenizer.decode, out))[0]
+    return generated.replace('\n[EOS]\n', '')
 
 
 medium_model, medium_tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_best_eval_loss.bin')
@@ -46,24 +47,25 @@ image = Image.open('korzh.jpg')
 st.image(image, caption='НейроКорж')
 
 option = st.selectbox('Выберите своего Коржа', ('Быстрый', 'Глубокий'))
-craziness = st.slider(label='Абсурдность', min_value=0.5, max_value=4., value=2.5, step=0.1)
+craziness = st.slider(label='Абсурдность', min_value=0, max_value=100, value=50, step=5)
+temperature = 2 + craziness / 50.
 
 st.markdown("\n")
 
-text = st.text_area(label='Напишите начало песни', value='Что делать, Макс?', height=100)
+text = st.text_area(label='Напишите начало песни', value='Что делать, Макс?', height=75)
 button = st.button('Старт')
 
 if button:
     try:
         with st.spinner("Пушечка пишется"):
             if option == 'Быстрый':
-                result = predict(text, medium_model, medium_tokenizer, temperature=craziness)
+                result = predict(text, medium_model, medium_tokenizer, temperature=temperature)
             elif option == 'Глубокий':
-                result = predict(text, large_model, large_tokenizer, temperature=craziness)
+                result = predict(text, large_model, large_tokenizer, temperature=temperature)
             else:
-                raise st.error('Error in selectbox')
+                st.error('Error in selectbox')
         
-        st.text_area(label='', value=result, height=1100)
+        st.text_area(label='', value=result, height=1000)
     
     except Exception:
         st.error("Ooooops, something went wrong. Please try again and report to me, tg: @vladyur")