AdityaQB commited on
Commit
dcecae7
·
verified ·
1 Parent(s): 9cce3d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, TFAutoModelForCausalLM
3
 
4
- def get_response(name):
5
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
6
  model = TFAutoModelForCausalLM.from_pretrained("gpt2")
7
  model.config.pad_token_id = model.config.eos_token_id
8
- inputs = tokenizer(["TensorFlow is"], return_tensors="tf")
9
 
10
  generated = model.generate(**inputs, do_sample=True, seed=(42, 0), max_new_tokens=20, temperature=.1)
11
  return tokenizer.decode(generated[0])
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, TFAutoModelForCausalLM
3
 
4
+ def get_response(name,text):
5
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
6
  model = TFAutoModelForCausalLM.from_pretrained("gpt2")
7
  model.config.pad_token_id = model.config.eos_token_id
8
+ inputs = tokenizer([text], return_tensors="tf")
9
 
10
  generated = model.generate(**inputs, do_sample=True, seed=(42, 0), max_new_tokens=20, temperature=.1)
11
  return tokenizer.decode(generated[0])