AbstractPhil commited on
Commit
40292d5
·
1 Parent(s): 644faa4
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -251,6 +251,7 @@ else:
251
  try:
252
  if seed is not None:
253
  torch.manual_seed(int(seed))
 
254
 
255
  # Load model
256
  model = _load_model_on("auto" if torch.cuda.is_available() else "cpu")
 
251
  try:
252
  if seed is not None:
253
  torch.manual_seed(int(seed))
254
+
255
 
256
  # Load model
257
  model = _load_model_on("auto" if torch.cuda.is_available() else "cpu")