MossaabDev commited on
Commit
a65f8bb
·
verified ·
1 Parent(s): 012edb0

Update app/data_loader.py

Browse files
Files changed (1) hide show
  1. app/data_loader.py +1 -1
app/data_loader.py CHANGED
@@ -10,7 +10,7 @@ print("Loading model and data...")
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
  print(f"Using device: {device}")
12
 
13
- model = SentenceTransformer("MossaabDev/quran_embed_V2", device=device)
14
  print("model size : ", asizeof.asizeof(model))
15
  df = pd.read_csv("app/data/cleaned_file.csv")
16
 
 
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
  print(f"Using device: {device}")
12
 
13
+ model = SentenceTransformer("app/my_finetuned_modelV2", device=device)
14
  print("model size : ", asizeof.asizeof(model))
15
  df = pd.read_csv("app/data/cleaned_file.csv")
16