3v324v23 commited on
Commit
ae0b701
1 Parent(s): 6fc3f73
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def generate_story(input_text):
16
  do_sample=True,
17
  top_k=40,
18
  top_p=0.9,
19
- temperature=0.6
20
  )
21
  return tokenizer.decode(output_sequences[0], skip_special_tokens=True)
22
 
@@ -30,7 +30,7 @@ interface = Interface(
30
  inputs=input_component,
31
  outputs="textbox",
32
  title="TinyStories-656K",
33
- description="This is a text generation model trained on a dataset of short stories. It can generate short stories based on your input.",
34
  examples=[['Once upon a time, there was a girl '], ['Long time ago, ']],
35
  theme="gradio/light"
36
  )
 
16
  do_sample=True,
17
  top_k=40,
18
  top_p=0.9,
19
+ temperature=0.65
20
  )
21
  return tokenizer.decode(output_sequences[0], skip_special_tokens=True)
22
 
 
30
  inputs=input_component,
31
  outputs="textbox",
32
  title="TinyStories-656K",
33
+ description="Try it!\nNote: Most of the time the default beginning works well.",
34
  examples=[['Once upon a time, there was a girl '], ['Long time ago, ']],
35
  theme="gradio/light"
36
  )