Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import random
|
|
3 |
import spaces
|
4 |
import os
|
5 |
|
6 |
-
os.environ["KERAS_BACKEND"] =
|
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
|
|
|
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()
|