Spaces:
Runtime error
Runtime error
File size: 565 Bytes
9cce3d7 a3d54b4 9cce3d7 f591b3e 9cce3d7 dcecae7 9cce3d7 c8abf63 9cce3d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import gradio as gr
from transformers import AutoTokenizer, TFAutoModelForCausalLM
def get_response(text):
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = TFAutoModelForCausalLM.from_pretrained("gpt2")
model.config.pad_token_id = model.config.eos_token_id
inputs = tokenizer([text], return_tensors="tf")
generated = model.generate(**inputs, do_sample=True, seed=(42, 0), max_new_tokens=20, temperature=.1)
return tokenizer.decode(generated[0])
iface = gr.Interface(fn=get_response, inputs="text", outputs="text")
iface.launch() |