egosumkira commited on
Commit
fa250ff
1 Parent(s): d6c38d2

add expanded desc

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -20,11 +20,11 @@ def generate(tags_text, temp, n_beams, max_len):
20
  return g_text[g_text.find("@") + 1:]
21
 
22
 
23
- title = "GPT2 fantasy story generator"
24
- description = "This model can generate short fantasy story based on set of keywords and (optional) start of the text."
25
-
26
- title = "GPT2 fantasy story generator"
27
- description = "This model can generate short fantasy story based on set of keywords and (optional) start of the text."
28
 
29
  iface = gr.Interface(generate,
30
  inputs = [
 
20
  return g_text[g_text.find("@") + 1:]
21
 
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 probabilities to sample other but the class with the highest log probability will be small, and the model will probably output the most correct text, but rather boring, with small variation. If the temperature is high, the model can output, with rather high probability, other words than those with the highest probability. 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 a 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 = [