Update models/prediction_sinhala.py

models/prediction_sinhala.py

@@ -183,7 +183,7 @@ from faknow.evaluate.evaluator import Evaluator
 
 import torch
 from torch.utils.data import DataLoader
-testing_path = "/content/drive/MyDrive/sinhala-dataset/test_data.json"
+testing_path = "data/test_data.json"
 
 df = pd.read_json(testing_path)
 df.head()
@@ -191,11 +191,10 @@ df =df[:100]
 df["label"] = int(0)
 df.head()
 print(len(df))
-path = '/content/drive/MyDrive/sinhala-dataset'
-testing_json = "/testing.json"
-df.to_json(path + testing_json, orient='records')
+new_testing_json_path = "data/testing.json"
+df.to_json(new_testing_json_path, orient='records')
 
-MODEL_SAVE_PATH = "/content/drive/MyDrive/models-path-improvement/last-epoch-model-2024-03-08-15_34_03_6.pth"
+MODEL_SAVE_PATH = "models/last-epoch-model-2024-03-08-15_34_03_6.pth"
 
 max_len, bert = 160 , 'sinhala-nlp/sinbert-sold-si'
 tokenizer = TokenizerFromPreTrained(max_len, bert)