Changed buttons to only show generations

app.py

@@ -83,7 +83,7 @@ def update_options(input, num_tokens):
                                             penalty=200)
     print("Generated")
     gens = alive_gens[0].reshape(n_drafts, -1)
-    return decode(tokenizer, gens[0]), decode(tokenizer, gens[1]), decode(tokenizer, gens[2])
+    return decode(tokenizer, gens[0])[len(input):], decode(tokenizer, gens[1])[len(input):], decode(tokenizer, gens[2])[len(input):]
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(