Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
|
|
|
|
|
|
|
|
|
| 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()
|