Update app.py

app.py

@@ -37,7 +37,7 @@ m.eval()
 #print(
 #    "Model with {:.2f}M parameters".format(sum(p.numel() for p in m.parameters()) / 1e6)
 #)
-def model_generate(text):
+def model_generate(text, number):
     # generate some output based on the context
     #context = torch.tensor(np.array(encode("Hello! My name is ", tokenizer)))
     #context = torch.zeros((1, 1), dtype=torch.long, device=DEVICE)
@@ -47,7 +47,7 @@ def model_generate(text):
     context = torch.from_numpy(context_np)
     #print(context)
 
-    return decode(enc_sec=m.generate(idx=context, max_new_tokens=100, block_size=BLOCK_SIZE)[0], tokenizer=tokenizer)
+    return decode(enc_sec=m.generate(idx=context, max_new_tokens=number, block_size=BLOCK_SIZE)[0], tokenizer=tokenizer)
 
-iface = gr.Interface(fn=model_generate, inputs="text", outputs="text")
+iface = gr.Interface(fn=model_generate, inputs=["text", gr.Slider(10, 1000)], outputs="text")
 iface.launch()