Update app.py

app.py

@@ -7,9 +7,9 @@ tokenizer = AutoTokenizer.from_pretrained("anonymous-german-nlp/german-gpt2")
 
 @st.cache
 def load_model(model_name):
-    model = AutoModelWithLMHead.from_pretrained("Jipski/Flos_gpt-2")
+    model = AutoModelWithLMHead.from_pretrained("Jipski/Flos_gpt-2_erw")
     return model
-model = load_model("Jipski/Flos_gpt-2")
+model = load_model("Jipski/Flos_gpt-2_erw")
 def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):
     output_sequences = model.generate(
         input_ids=input_ids,