TitleOS commited on
Commit
e8db596
·
1 Parent(s): 9b8f7c9

Create app.py

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