Younes Belkada commited on
Commit
fed3237
1 Parent(s): 7012779

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -31,5 +31,18 @@ def inference(text, seq_length=1):
31
  input_ids = next_token_ix.view(1, 1)
32
  return tokenizer.decode(final_tokens[0], skip_special_tokens=False)
33
 
34
- iface = gr.Interface(fn=inference, inputs="text", outputs="text")
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  iface.launch()
 
31
  input_ids = next_token_ix.view(1, 1)
32
  return tokenizer.decode(final_tokens[0], skip_special_tokens=False)
33
 
34
+ iface = gr.Interface(
35
+ fn=inference,
36
+ inputs=[
37
+ gr.Textbox(lines=10, label="Input text"),
38
+ gr.inputs.Slider(
39
+ minimum=0,
40
+ maximum=1000,
41
+ step=1,
42
+ default=42,
43
+ label="Sequence length for generation"
44
+ )
45
+ ],
46
+ outputs="text"
47
+ )
48
  iface.launch()