Update app.py

app.py

@@ -38,9 +38,11 @@ m.eval()
 #    "Model with {:.2f}M parameters".format(sum(p.numel() for p in m.parameters()) / 1e6)
 #)
 def model_generate(text, number_of_new_token, temperature, top_p):
+    print(text)
     context = encode(str(text), tokenizer).unsqueeze(0).to(DEVICE)
 
     gen = decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p)[0], tokenizer=tokenizer)
+    print(gen)
     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")