dwb2023 commited on
Commit
01ebc72
·
verified ·
1 Parent(s): 0fdb67e

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -0
inference.py CHANGED
@@ -7,6 +7,7 @@ import jax
7
  import jax.numpy as jnp
8
  import numpy as np
9
  import functools
 
10
 
11
  hf_token = os.getenv("HF_TOKEN")
12
  login(token=hf_token, add_to_git_credential=True)
 
7
  import jax.numpy as jnp
8
  import numpy as np
9
  import functools
10
+ import spaces
11
 
12
  hf_token = os.getenv("HF_TOKEN")
13
  login(token=hf_token, add_to_git_credential=True)