ariG23498 HF Staff commited on
Commit
76fae55
·
1 Parent(s): 2e9f593
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import keras_core as keras
3
  import gradio as gr
4
  from huggingface_hub import Repository
 
1
  import os
2
+ os.environ["KERAS_BACKEND"] = "jax"
3
  import keras_core as keras
4
  import gradio as gr
5
  from huggingface_hub import Repository
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  keras-core
2
  keras-nlp
3
  huggingface_hub
 
1
+ jax
2
+ jaxlib
3
  keras-core
4
  keras-nlp
5
  huggingface_hub