ARTPARK commited on
Commit
4871a18
·
verified ·
1 Parent(s): 7b7d2ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ print("Loading model, this may take a few minutes...")
11
  model = EncDecRNNTBPEModel.from_pretrained(MODEL_NAME)
12
  model.eval()
13
 
14
- # Use CPU if GPU is not available
15
  if not torch.cuda.is_available():
16
  model = model.cpu()
17
  print("Model loaded successfully.")
 
11
  model = EncDecRNNTBPEModel.from_pretrained(MODEL_NAME)
12
  model.eval()
13
 
14
+ #Use CPU if GPU is not available
15
  if not torch.cuda.is_available():
16
  model = model.cpu()
17
  print("Model loaded successfully.")