Update app.py

app.py

@@ -32,12 +32,7 @@ def paraphrase(
     )
 
     res = tokenizer.batch_decode(outputs, skip_special_tokens=True)
-
-    result = ''
-    for i, item in enumerate(res):
-        result += f'{i+1}. {item}\n'
-
-    return result
+    return res
 
 def fn(
     text,
@@ -50,7 +45,11 @@ def fn(
     temperature=0.7,
     max_length=128
 ):
-    return '\n'.join(paraphrase(text, num_beams, num_beam_groups, num_return_sequences, repetition_penalty, diversity_penalty, no_repeat_ngram_size, temperature, max_length))
+    res = paraphrase(text, num_beams, num_beam_groups, num_return_sequences, repetition_penalty, diversity_penalty, no_repeat_ngram_size, temperature, max_length)
+    result = ''
+    for i, item in enumerate(res):
+        result += f'{i+1}. {item}\n'
+    return result
 
 demo = gr.Interface(
     fn=fn,