Commit
·
971afa1
1
Parent(s):
f9f05d9
correctly named hubert pth
Browse files
app.py
CHANGED
@@ -246,7 +246,7 @@ associated_links = {}
|
|
246 |
def load_hubert():
|
247 |
global hubert_model
|
248 |
# Load the model
|
249 |
-
hubert_model = torch.load("
|
250 |
|
251 |
# Prepare the model
|
252 |
hubert_model = hubert_model.to(config.device)
|
|
|
246 |
def load_hubert():
|
247 |
global hubert_model
|
248 |
# Load the model
|
249 |
+
hubert_model = torch.load("hubert.pth", map_location=config.device)
|
250 |
|
251 |
# Prepare the model
|
252 |
hubert_model = hubert_model.to(config.device)
|