Update app.py
Browse files
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 |
-
#
|
| 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.")
|