azaninello commited on
Commit
3172561
·
1 Parent(s): b255655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -10,13 +10,13 @@ tokenizer = GPT2Tokenizer.from_pretrained(
10
 
11
  generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('gpt2'),
12
  tokenizer='gpt2',
13
- do_sample=True,
14
- max_length=120,
15
- top_k=50,
16
- top_p=0.95,)
17
 
18
  def generator(start_your_text = ''):
19
- shroom_result = generator(start_your_text, max_length=120)
20
  return shroom_result[0]["generated_text"]
21
 
22
  iface = gr.Interface(fn=generator, inputs="text", outputs="text")
 
10
 
11
  generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('gpt2'),
12
  tokenizer='gpt2',
13
+ #do_sample=True,
14
+ #max_length=120,
15
+ #top_k=50,
16
+ #top_p=0.95,)
17
 
18
  def generator(start_your_text = ''):
19
+ shroom_result = generator(start_your_text)
20
  return shroom_result[0]["generated_text"]
21
 
22
  iface = gr.Interface(fn=generator, inputs="text", outputs="text")