Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_text(seed_text, max_new_tokens, confidence_val):
     text = text if text.endswith(" ") else seed_text + " "
     context = torch.tensor(encode(text), dtype=torch.long).unsqueeze(0)
     confidence_val = confidence_val if confidence_val > 0 else 1e-5
-    return decode(trained_model.generate(context, confidence_val = confidence_val, max_new_tokens=max_new_tokens)[0].tolist())
+    return decode(trained_model.generate(context, temperature = confidence_val, max_new_tokens=max_new_tokens)[0].tolist())
 
 with gr.Blocks() as demo:
     gr.HTML("<h1 align = 'center'> Simple GPT from scratch using tiny Shakespere </h1>")