Hugging Face's logo

Spaces:
LuyangZ
/
FloorAI
Runtime error

App Files Community
1
LuyangZ commited on
Commit
ab43ad3
·
verified ·
1 Parent(s): eed3e5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,7 +12,9 @@ import transformers
12
  from random import randrange
13
 
14
 
15
- # transformers.utils.move_cache()
 
 
16
 
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
 
 
12
  from random import randrange
13
 
14
 
15
+ from transformers.utils.hub import move_cache
16
+ move_cache()
17
+
18
 
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20