Spaces:
Runtime error
Runtime error
Update app/data_loader.py
Browse files- 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("
|
| 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 |
|