Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
| 8 |
response = generated_text.split("### Response")[-1]
|
| 9 |
return response
|
| 10 |
|
| 11 |
interface = gr.Interface(
|
| 12 |
fn=generate_text,
|
| 13 |
inputs=[
|
| 14 |
-
gr.
|
| 15 |
-
gr.
|
| 16 |
],
|
| 17 |
-
outputs=gr.
|
| 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()
|