Spaces:
Runtime error
Runtime error
NegativeSector
commited on
Commit
•
9f469ad
1
Parent(s):
73d474e
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,18 @@
|
|
1 |
import gradio as gr
|
2 |
import json
|
3 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
4 |
-
|
5 |
tokenizer = AutoTokenizer.from_pretrained("gpt2")
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
model = AutoModelWithLMHead.from_pretrained("gpt2")
|
8 |
def green(prompt):
|
9 |
-
|
|
|
10 |
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
|
11 |
iface.launch()
|
|
|
1 |
import gradio as gr
|
2 |
import json
|
3 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
4 |
+
from transformers import pipeline
|
5 |
tokenizer = AutoTokenizer.from_pretrained("gpt2")
|
6 |
+
model = AutoModelWithLMHead.from_pretrained("gpt2-large")
|
7 |
+
generator = pipeline(
|
8 |
+
task="text-generation",
|
9 |
+
model=model,
|
10 |
+
tokenizer=tokenizer,
|
11 |
+
framework="tf"
|
12 |
+
)
|
13 |
model = AutoModelWithLMHead.from_pretrained("gpt2")
|
14 |
def green(prompt):
|
15 |
+
text_generator(prompt, max_length=500)
|
16 |
+
|
17 |
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
|
18 |
iface.launch()
|