Updated app

app.py

@@ -1,13 +1,11 @@
 import streamlit as st
-from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
+from transformers import AutoTokenizer, AutoModelWithLMHead, GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
 import torch
 
 #===========================================#
 #        Loads Model and Pipeline        #
 #===========================================#
 
-from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
- 
 tokenizer = AutoTokenizer.from_pretrained("flax-community/swe-gpt-wiki")
 model = AutoModelWithLMHead.from_pretrained("flax-community/swe-gpt-wiki")
 
@@ -18,13 +16,14 @@ set_seed(42)
 #===========================================#
 #              Streamlit Code               #
 #===========================================#
-desc = "En svensk GPT-modell tränad på wikipedia"
 
-st.title('Vi använder wikipidea för att generera text')
+st.title('Vi använder wikipidea för att generera text.')
+desc = "En svensk GPT-modell tränad på wikipedia. Fyll i text i fältet nedanför och hur mycket text du vill ska genereras."
+
 st.write(desc)
 
-num_sentences = st.number_input('Number of Sentences', min_value=1, max_value=20, value=5)
-user_input = st.text_input('Seed Text (can leave blank)')
+num_sentences = st.number_input('Number of Characters', min_value=1, max_value=150, value=30)
+user_input = st.text_input('Vi börjar med en text (can leave blank)')
 
 
 if st.button('Generate Text'):