Spaces:
Runtime error
Runtime error
File size: 373 Bytes
e9937d0 62d6ebb 1334a88 e9937d0 1334a88 a43ff95 e9937d0 |
1 2 3 4 5 6 7 8 9 10 11 |
import gradio as gr
import json
from transformers import AutoTokenizer, AutoModelWithLMHead
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelWithLMHead.from_pretrained("gpt2")
def green(prompt):
print(tokenizer("help"))
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
iface.launch() |