Spaces:
Sleeping
Sleeping
fix tokenizer_kwargs
Browse files
main.py
CHANGED
@@ -11,7 +11,7 @@ pipe = pipeline("text-generation", model="ai-forever/rugpt3large_based_on_gpt2")
|
|
11 |
|
12 |
@app.get("/infer_t5")
|
13 |
def t5(input):
|
14 |
-
tokenizer_kwargs = {'
|
15 |
output = pipe(input, **tokenizer_kwargs)
|
16 |
return {"output": output[0]["generated_text"]}
|
17 |
|
|
|
11 |
|
12 |
@app.get("/infer_t5")
|
13 |
def t5(input):
|
14 |
+
tokenizer_kwargs = {'max_length':512}
|
15 |
output = pipe(input, **tokenizer_kwargs)
|
16 |
return {"output": output[0]["generated_text"]}
|
17 |
|