alexdong commited on
Commit
d15a8c3
·
verified ·
1 Parent(s): c386046

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def generate_text(input_text):
11
  input_ids = tokenizer(f"reformulate: {input_text}", return_tensors="pt").input_ids
12
  output_ids = model.generate(input_ids, max_length=50)
13
  decoded_output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
14
- return decoded_output.split("\n")
15
 
16
  # Define the Gradio interface
17
  iface = gr.Interface(
 
11
  input_ids = tokenizer(f"reformulate: {input_text}", return_tensors="pt").input_ids
12
  output_ids = model.generate(input_ids, max_length=50)
13
  decoded_output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
14
+ return decoded_output.split("\\n")
15
 
16
  # Define the Gradio interface
17
  iface = gr.Interface(