Amitesh007 commited on
Commit
a3476e7
1 Parent(s): 0dcfdd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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=300
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()