Spaces:
Runtime error
Runtime error
Amitesh007
commited on
Commit
•
a3476e7
1
Parent(s):
0dcfdd0
Update app.py
Browse files
app.py
CHANGED
@@ -5,11 +5,13 @@ fine_tuned_model = "Amitesh007/text_generation-finetuned-gpt2"
|
|
5 |
def generate(text):
|
6 |
generator = pipeline('text-generation', model=fine_tuned_model)
|
7 |
result = generator(text, num_return_sequences=2,
|
8 |
-
max_length=
|
9 |
|
10 |
)
|
11 |
return result[0]["generated_text"],result[1]["generated_text"]
|
12 |
|
|
|
|
|
13 |
demo = gr.Interface(
|
14 |
fn=generate,
|
15 |
inputs=[gr.Textbox(lines=5, label="Input Text here....")],
|
@@ -19,7 +21,8 @@ demo = gr.Interface(
|
|
19 |
|
20 |
],
|
21 |
title = "Text Generator GPT2 Pipeline",
|
22 |
-
description = "this is a fine-tuned base gpt2 model inference, trained on a small 'game of thrones dataset
|
|
|
23 |
)
|
24 |
demo.queue(concurrency_count=2)
|
25 |
demo.launch()
|
|
|
5 |
def generate(text):
|
6 |
generator = pipeline('text-generation', model=fine_tuned_model)
|
7 |
result = generator(text, num_return_sequences=2,
|
8 |
+
max_length=100
|
9 |
|
10 |
)
|
11 |
return result[0]["generated_text"],result[1]["generated_text"]
|
12 |
|
13 |
+
Examples = [["A light snow had fallen the night before, and there were"],["The pig face had been smashed in with a mace, but Tyrion"]]
|
14 |
+
|
15 |
demo = gr.Interface(
|
16 |
fn=generate,
|
17 |
inputs=[gr.Textbox(lines=5, label="Input Text here....")],
|
|
|
21 |
|
22 |
],
|
23 |
title = "Text Generator GPT2 Pipeline",
|
24 |
+
description = "this is a fine-tuned base gpt2 model inference, trained on a small 'game of thrones' dataset",
|
25 |
+
examples = Examples
|
26 |
)
|
27 |
demo.queue(concurrency_count=2)
|
28 |
demo.launch()
|