umuthopeyildirim commited on
Commit
913f98e
1 Parent(s): 9c21a6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,7 +37,7 @@ def predict(message, history):
37
  generate_kwargs = dict(
38
  model_inputs,
39
  streamer=streamer,
40
- max_new_tokens=1024,
41
  do_sample=True,
42
  top_p=0.95,
43
  top_k=1000,
@@ -70,7 +70,7 @@ def generate_text(prompt, tokenizer, model):
70
 
71
  title = "# Fin-RWKV: Attention Free Financal Expert (WIP)"
72
  description = """Demo for **Fin-RWKV: Attention Free Financal Expert (WIP)**.
73
- To download the model, please visit [Fin-RWKV: Attention Free Financal Expert (WIP)](https://huggingface.co/umutyildirim/fin-rwkv-1b5)."""
74
 
75
  css = """
76
  #img-display-container {
 
37
  generate_kwargs = dict(
38
  model_inputs,
39
  streamer=streamer,
40
+ max_new_tokens=9999,
41
  do_sample=True,
42
  top_p=0.95,
43
  top_k=1000,
 
70
 
71
  title = "# Fin-RWKV: Attention Free Financal Expert (WIP)"
72
  description = """Demo for **Fin-RWKV: Attention Free Financal Expert (WIP)**.
73
+ To download the model, please visit [Fin-RWKV: Attention Free Financal Expert (WIP)](https://huggingface.co/umuthopeyildirim/fin-rwkv-1b5)."""
74
 
75
  css = """
76
  #img-display-container {