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

Update app.py

Browse files

Bigger prompt and answer?

Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -1,3 +1,16 @@
1
  import gradio as gr
 
2
 
3
- gr.Interface.load("models/EleutherAI/gpt-neo-1.3B").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
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()