Update app.py

app.py

@@ -41,7 +41,7 @@ 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)
+    gen = decode(enc_sec=m.generate(idx=context, max_new_tokens=number_of_new_token, temperature = temperature, top_p = top_p), tokenizer=tokenizer)
     print(gen)
     return gen