egosumkira commited on
Commit
c54c05f
1 Parent(s): b322734

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -22,10 +22,8 @@ def generate(tags_text, temp, n_beams, max_len):
22
 
23
  title = "GPT-2 fantasy story generator"
24
  description = 'This is fine-tuned GPT-2 model for "conditional" generation. The model was trained on a custom-made dataset of IMDB plots & keywords.\n' \
25
- 'There are two main parameters to generate output:\n' \
26
- '1. Temperature. If the temperature is low, the model will probably output the most correct text, but rather boring, with small variation. If the temperature is high, the generated text will be more diverse, but there is a higher possibility of grammar mistakes and generation of nonsense.\n'\
27
- '2. Number of beams in Beam Search. Beam search is a clever way to find the best sentences in an algorithm that writes words. It looks at a few possible sentences at a time, and keeps track of the most promising ones.'
28
-
29
  iface = gr.Interface(generate,
30
  inputs = [
31
  gr.Textbox(label="Keywords (comma separated)"),
 
22
 
23
  title = "GPT-2 fantasy story generator"
24
  description = 'This is fine-tuned GPT-2 model for "conditional" generation. The model was trained on a custom-made dataset of IMDB plots & keywords.\n' \
25
+ 'Model page: https://huggingface.co/egosumkira/gpt2-fantasy \n' \
26
+ 'Notebooks: https://github.com/Agniwald/GPT-2-Fantasy'
 
 
27
  iface = gr.Interface(generate,
28
  inputs = [
29
  gr.Textbox(label="Keywords (comma separated)"),