tatihden commited on
Commit
6d7f46d
·
verified ·
1 Parent(s): 2bb5246

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import random
3
  import spaces
4
  import os
5
 
6
- os.environ["KERAS_BACKEND"] = "jax"
7
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
8
 
9
  import keras_hub
@@ -107,4 +107,4 @@ with gr.Blocks(css=".gradio-container {background-color: rgb(187 247 208)}") as
107
  top_k], chat_b)
108
  stop_btn.click(None, None, None, cancels=[go, chat_sub])
109
  clear_btn.click(clear_fn, None, [chat_b])
110
- app.queue(default_concurrency_limit=10,type="messages").launch()
 
3
  import spaces
4
  import os
5
 
6
+ os.environ["KERAS_BACKEND"] = "torch" #"jax"
7
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
8
 
9
  import keras_hub
 
107
  top_k], chat_b)
108
  stop_btn.click(None, None, None, cancels=[go, chat_sub])
109
  clear_btn.click(clear_fn, None, [chat_b])
110
+ app.queue(default_concurrency_limit=10).launch()