Update app.py

app.py

@@ -1,11 +1,13 @@
 import transformers
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelWithLMHead
-  
-tokenizer = AutoTokenizer.from_pretrained("anonymous-german-nlp/german-gpt2")
+
+tokenizer = AutoTokenizer.from_pretrained("dbmdz/german-gpt2")
+
+
 @st.cache
 def load_model(model_name):
-    model = "Jipski/Flos_gpt-2"
+    model = AutoModelWithLMHead.from_pretrained("Jipski/Flos_gpt-2")
     return model
 model = load_model("Jipski/Flos_gpt-2")
 def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):