Update app.py

app.py

@@ -1,15 +1,13 @@
 import transformers
 import streamlit as st
-
 from transformers import AutoTokenizer, AutoModelWithLMHead
-
-tokenizer='anonymous-german-nlp/german-gpt2'
-
+  
+tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
 @st.cache
 def load_model(model_name):
-    model = AutoModelWithLMHead.from_pretrained("Flos_gpt-2")
+    model = AutoModelWithLMHead.from_pretrained("gpt2-large")
     return model
-model = load_model("Flos_gpt-2")
+model = load_model("gpt2-large")
 def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):
     output_sequences = model.generate(
         input_ids=input_ids,
@@ -50,4 +48,4 @@ for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
     )
     generated_sequences.append(total_sequence)
     print(total_sequence)
-st.write(generated_sequences[-1])
+st.write(generated_sequences[-1])