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()