reach-vb HF staff commited on
Commit
58aa497
1 Parent(s): c17c03f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,7 +33,10 @@ def pred(text_in):
33
  text_out = tokenizer.batch_decode(out.sequences.tolist())
34
  return text_out[0]
35
 
36
- demo = gr.Interface(fn=pred, inputs="text", outputs="text")
 
 
 
37
 
38
  if __name__ == "__main__":
39
  demo.launch()
 
33
  text_out = tokenizer.batch_decode(out.sequences.tolist())
34
  return text_out[0]
35
 
36
+ demo = gr.Interface(
37
+ title="Mamba: Selective State Space Model",
38
+ description="A demo for [Mamba](https://github.com/state-spaces/mamba)",
39
+ fn=pred, inputs="text", outputs="text")
40
 
41
  if __name__ == "__main__":
42
  demo.launch()