Spaces:
Runtime error
Runtime error
egosumkira
commited on
Commit
•
c54c05f
1
Parent(s):
b322734
Update app.py
Browse files
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 |
-
'
|
26 |
-
'
|
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)"),
|