TitleOS commited on
Commit
488be40
·
verified ·
1 Parent(s): b7714f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,17 +4,17 @@ from transformers import pipeline
4
 
5
  def generate_text(prompt, temperature):
6
  generator = pipeline('text-generation', model='titleos/Seahorse-350m')
7
- generated_text = generator("### Instruction " + prompt + " ### Response ", max_length=100, temperature=temperature)[0]['generated_text']
8
  response = generated_text.split("### Response")[-1]
9
  return response
10
 
11
  interface = gr.Interface(
12
  fn=generate_text,
13
  inputs=[
14
- gr.inputs.Textbox(lines=1, label="Instruction", default="What are the three primary colors?"),
15
- gr.inputs.Slider(minimum=0.1, maximum=1.0, step=0.1, default=0.5, label="Temperature")
16
  ],
17
- outputs=gr.outputs.Textbox(label="Response")
18
  )
19
 
20
  interface.launch()
 
4
 
5
  def generate_text(prompt, temperature):
6
  generator = pipeline('text-generation', model='titleos/Seahorse-350m')
7
+ generated_text = generator("### Instruction " + prompt + " ### Response ", max_length=250, temperature=temperature)[0]['generated_text']
8
  response = generated_text.split("### Response")[-1]
9
  return response
10
 
11
  interface = gr.Interface(
12
  fn=generate_text,
13
  inputs=[
14
+ gr.Textbox(lines=3, label="Instruction", placeholder="What are the three primary colors?"),
15
+ gr.Slider(minimum=0.1, maximum=2.0, step=0.1, default=1.0, label="Temperature")
16
  ],
17
+ outputs=gr.Textbox(label="Response")
18
  )
19
 
20
  interface.launch()