crystal99 commited on
Commit
5f9b31b
·
verified ·
1 Parent(s): c3b57aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
10
  def generate_text(prompt):
11
  inputs = tokenizer(prompt, return_tensors="pt")
12
  outputs = model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1)
13
- generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
14
  return generated_text
15
 
16
  # Set up the Gradio interface
 
10
  def generate_text(prompt):
11
  inputs = tokenizer(prompt, return_tensors="pt")
12
  outputs = model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1)
13
+ generated_text = tokenizer.decode(outputs[0], skip_special_tokens=False)
14
  return generated_text
15
 
16
  # Set up the Gradio interface