Hugging Face's logo

Spaces:
Jipski
/
Flos_gpt-2
Runtime error

App Files Community
Jipski commited on
Commit
4161351
·
1 Parent(s): 835aa93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ tokenizer = AutoTokenizer.from_pretrained("anonymous-german-nlp/german-gpt2")
7
 
8
  @st.cache
9
  def load_model(model_name):
10
- model = AutoModelWithLMHead.from_pretrained("Jipski/Flos_gpt-2_erw")
11
  return model
12
  model = load_model("Jipski/Flos_gpt-2_erw")
13
  def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):
 
7
 
8
  @st.cache
9
  def load_model(model_name):
10
+ model = AutoModelWithLMHead.from_pretrained("Jipski/Flos_gpt-2_erw-02")
11
  return model
12
  model = load_model("Jipski/Flos_gpt-2_erw")
13
  def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):