lachine commited on
Commit
5bd8055
1 Parent(s): a5d3681

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -17
app.py CHANGED
@@ -1,18 +1,2 @@
1
- import gradio as gr
2
- from transformers import pipeline
3
  from os import system
4
- system('pip install transformers')
5
- text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
6
-
7
- def generate_text(prompt, quality):
8
- length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}
9
- length = length_dict[quality]
10
- return text_generator(prompt, max_length=length, do_sample=True)[0]['generated_text']
11
-
12
- iface = gr.Interface(
13
- fn=generate_text,
14
- inputs=["textbox", gr.inputs.Dropdown(["Low", "Medium (Default)", "High"], label="Quality")],
15
- outputs=gr.outputs.Image(label="Generated image:")
16
- )
17
-
18
- iface.launch()
 
 
 
1
  from os import system
2
+ system('pip install transformers')