eskayML commited on
Commit
64ca7f4
1 Parent(s): 318a832

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = AutoModelForCausalLM.from_pretrained("distilgpt2")
7
 
8
 
9
  gen_pipeline = pipeline(task = "text-generation", model=model, tokenizer=tokenizer)
10
- get_generated_text = lambda x: gen_pipeline(x)[0]['generated_text']
11
 
12
 
13
  demo = gr.Interface(
 
7
 
8
 
9
  gen_pipeline = pipeline(task = "text-generation", model=model, tokenizer=tokenizer)
10
+ get_generated_text = lambda x: gen_pipeline(x,max_length = 200)[0]['generated_text']
11
 
12
 
13
  demo = gr.Interface(