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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ 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
+ print(decoded_output)
15
+ return decoded_output
16
 
17
  # Define the Gradio interface
18
  iface = gr.Interface(