Hugging Face's logo

Spaces:
Ashrafb
/
VT4
Running

App Files Community
Ashrafb commited on
Commit
eff2d0e
·
verified ·
1 Parent(s): add003d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -34,6 +34,7 @@ import huggingface_hub
34
  import os
35
 
36
  app = FastAPI()
 
37
 
38
 
39
 
 
34
  import os
35
 
36
  app = FastAPI()
37
+ model = Model(device='cuda' if torch.cuda.is_available() else 'cpu')
38
 
39
 
40