Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,9 @@ 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-
|
11 |
return model
|
12 |
-
model = load_model("Jipski/Flos_gpt-
|
13 |
def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences):
|
14 |
output_sequences = model.generate(
|
15 |
input_ids=input_ids,
|
|
|
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):
|
14 |
output_sequences = model.generate(
|
15 |
input_ids=input_ids,
|