Hugging Face's logo

Spaces:
oucgc1996
/
DeepACE
Sleeping

App Files Community
oucgc1996 commited on
Commit
cff9fff
·
verified ·
1 Parent(s): 0a3a5fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
 
11
  warnings.filterwarnings('ignore')
12
  set_seed(4)
13
- device = "cuda:0"
14
  model_checkpoint = "facebook/esm2_t30_150M_UR50D"
15
 
16
  class MyModel(nn.Module):
 
10
 
11
  warnings.filterwarnings('ignore')
12
  set_seed(4)
13
+ device = "cpu"
14
  model_checkpoint = "facebook/esm2_t30_150M_UR50D"
15
 
16
  class MyModel(nn.Module):