picocreator commited on
Commit
69c97e8
·
1 Parent(s): a387c45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -14
app.py CHANGED
@@ -1,16 +1,3 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
 
4
- # gr.Interface.load("models/EleutherAI/gpt-neo-1.3B").launch()
5
-
6
- model_name = "models/EleutherAI/gpt-neo-1.3B"
7
- generator = pipeline('text-generation', model = model_name)
8
-
9
- def generate_text(text_prompt):
10
- response = generator(text_prompt, max_length = 200, num_return_sequences=10)
11
- return response[0]['generated_text']
12
-
13
- textbox = gr.Textbox()
14
-
15
- demo = gr.Interface(generate_text, textbox, textbox)
16
- demo.launch()
 
1
  import gradio as gr
 
2
 
3
+ gr.Interface.load("models/EleutherAI/gpt-neo-1.3B").launch()