spuun commited on
Commit
46c38c7
1 Parent(s): 4711ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import gradio as gr
11
 
12
  def generate(input):
13
  tokenized = tokenizer(input, return_tensors="pt")["input_ids"]
14
- outputs = model.generate(inputs, max_new_tokens=80, do_sample=True, temperature=0.9)
15
  return tokenizer.decode(outputs[0]).replace("</s>", "");
16
 
17
  iface = gr.Interface(fn=generate, inputs="text", outputs="text")
 
11
 
12
  def generate(input):
13
  tokenized = tokenizer(input, return_tensors="pt")["input_ids"]
14
+ outputs = model.generate(tokenized, max_new_tokens=80, do_sample=True, temperature=0.9)
15
  return tokenizer.decode(outputs[0]).replace("</s>", "");
16
 
17
  iface = gr.Interface(fn=generate, inputs="text", outputs="text")