update base url

app.py

@@ -1,3 +1,4 @@
+import os
 import torch
 import argparse
 import sentencepiece as spm
@@ -5,8 +6,8 @@ from utils import utils_cls
 from model import BanglaTransformer
 from config import config as cfg
 torch.manual_seed(0)
-# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-device = torch.device('cpu')
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# device = torch.device('cpu')
 uobj = utils_cls(device=device)
 
 __MODULE__       = "Bangla Language Translation"
@@ -14,15 +15,15 @@ __MAIL__         = "saifulbrur79@gmail.com"
 __MODIFICAIOTN__ = "28/03/2023"
 __LICENSE__      = "MIT"
 
-
+BASE_URL = "https://huggingface.co/saiful9379/Bangla-to-English-Translation/tree/main"
 
 class Bn2EnTranslation:
     def __init__(self):
-        self.bn_tokenizer='./model/bn_model.model'
-        self.en_tokenizer='./model/en_model.model'
-        self.bn_vocab='./model/bn_vocab.pkl'
-        self.en_vocab='./model/en_vocab.pkl'
-        self.model='./model/pytorch_model.pt'
+        self.bn_tokenizer= os.path.join(BASE_URL , "bn_model.model")
+        self.en_tokenizer=os.path.join(BASE_URL, 'en_model.model')
+        self.bn_vocab=os.path.join(BASE_URL,'bn_vocab.pkl')
+        self.en_vocab=os.path.join(BASE_URL, 'en_vocab.pkl')
+        self.model= os.path.join(BASE_URL,'pytorch_model.pt')
 
     def read_data(self, data_path):
         with open(data_path, "r") as f: