Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ import gradio as gr
|
|
28 |
|
29 |
device = 'cpu'
|
30 |
le = LabelEncoder()
|
31 |
-
le = joblib.load("
|
32 |
|
33 |
class ModelPre(torch.nn.Module):
|
34 |
def __init__(self):
|
@@ -44,7 +44,7 @@ class ModelPre(torch.nn.Module):
|
|
44 |
def forward(self, data):
|
45 |
return self.embedding(data)
|
46 |
|
47 |
-
model = torch.load("
|
48 |
|
49 |
modelm = ModelPre()
|
50 |
modelm.load_state_dict(model['model'])
|
|
|
28 |
|
29 |
device = 'cpu'
|
30 |
le = LabelEncoder()
|
31 |
+
le = joblib.load("~/SVD/le.gz")
|
32 |
|
33 |
class ModelPre(torch.nn.Module):
|
34 |
def __init__(self):
|
|
|
44 |
def forward(self, data):
|
45 |
return self.embedding(data)
|
46 |
|
47 |
+
model = torch.load("~/SVD/GeoG.pth", map_location=torch.device(device))
|
48 |
|
49 |
modelm = ModelPre()
|
50 |
modelm.load_state_dict(model['model'])
|