Spaces:

shandong1970
/

list5-11

Sleeping

shandong1970 commited on May 27

Commit

a3e4c46

•

1 Parent(s): 4dc84dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ gpt2_tkn = GPT2Tokenizer.from_pretrained("gpt2")
 def generate(starting_text):
     tkn_ids = gpt2_tkn.encode(starting_text,return_tensors='pt')
-    gpt2_tensors = mdl.generate(tkn_ids, max_length=100, no_repeat_ngram_size=True,num_beams=3,do_sample=True,temperature=1.5)
     response = ""
     #response = gpt2_tensors
     for i, x in enumerate(gpt2_tensors):

 def generate(starting_text):
     tkn_ids = gpt2_tkn.encode(starting_text,return_tensors='pt')
+    gpt2_tensors = mdl.generate(tkn_ids, max_length=100, no_repeat_ngram_size=True,num_beams=3,do_sample=True,temperature=0.1)
     response = ""
     #response = gpt2_tensors
     for i, x in enumerate(gpt2_tensors):