Update app.py

app.py

@@ -40,7 +40,8 @@ m.eval()
 def model_generate(text, number_of_new_token, temperature, top_p):
     context = encode(str(text), tokenizer).unsqueeze(0).to(DEVICE)
 
-    return decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
+    gen = decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
+    return gen
 
 iface = gr.Interface(fn=model_generate, inputs=["text", gr.Slider(10, 1000), gr.Slider(0, 1, value=0.7, step = 0.05), gr.Slider(0, 1, value=0.95, step = 0.05)], outputs="text")
 iface.launch()