Updated app

app.py

@@ -6,10 +6,10 @@ import torch
 #        Loads Model and Pipeline        #
 #===========================================#
 
-tokenizer = AutoTokenizer.from_pretrained("flax-community/swe-gpt-wiki")
-model = AutoModelWithLMHead.from_pretrained("flax-community/swe-gpt-wiki")
+# tokenizer = AutoTokenizer.from_pretrained("flax-community/swe-gpt-wiki")
+# model = AutoModelWithLMHead.from_pretrained("flax-community/swe-gpt-wiki")
 
-generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
+# generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 # set_seed(42)
 
 
@@ -21,9 +21,20 @@ generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 #     "What language model would you like to try?",
 #     ("Swedish", "Norwegian", "Danish", "Nordic"))
 
+gpt_models = ["birgermoell/swedish-gpt", "flax-community/swe-gpt-wiki", "flax-community/nordic-gpt-wiki", "flax-community/norsk-gpt-wiki", "flax-community/dansk-gpt-wiki"]
+
+selected_model = st.selectbox(
+    "What language model would you like to try?",
+    (gpt_models))
+
+tokenizer = AutoTokenizer.from_pretrained(selected_model)
+model = AutoModelWithLMHead.from_pretrained(selected_model)
+
+generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
+
 
-st.title('Svensk GPT text generering')
-desc = "En svensk GPT-modell tränad på wikipedia. Fyll i text i fältet nedanför för generering."
+st.title('GPT text generering')
+desc = "Pröva GPT-modeller på flera språk. Skriv text nedanför för att utvärdera. Använder " + selected_model + " för att generera text. " + "Här kan du läsa mer och modellen https://huggingface.co/" + selected_model
 
 st.write(desc)