Spaces:

woshixuhao
/

Rf_prediction

Running

woshixuhao commited on May 11, 2023

Commit

fb4cfa0

1 Parent(s): 7980d1c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ def parse_args():
     config = parser.parse_args()
-    config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
     return config
 class ANN(nn.Module):

     config = parser.parse_args()
+    config.device = 'cpu'
     return config
 class ANN(nn.Module):