MossaabDev commited on
Commit
eebabd3
·
verified ·
1 Parent(s): fa2313a

Update app/data_loader.py

Browse files
Files changed (1) hide show
  1. app/data_loader.py +1 -1
app/data_loader.py CHANGED
@@ -12,7 +12,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
12
  print(f"Using device: {device}")
13
 
14
  # --- Load model ---
15
- model = SentenceTransformer("MossaabDev/Quran_embed_V2", device=device)
16
  print("Model size:", asizeof.asizeof(model))
17
 
18
  # --- Load ayahs from ayas.csv ---
 
12
  print(f"Using device: {device}")
13
 
14
  # --- Load model ---
15
+ model = SentenceTransformer("MossaabDev/Quran_embed_V2.2", device=device)
16
  print("Model size:", asizeof.asizeof(model))
17
 
18
  # --- Load ayahs from ayas.csv ---