robocan commited on
Commit
f40280d
·
verified ·
1 Parent(s): b17926f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,7 +28,7 @@ import gradio as gr
28
 
29
  device = 'cpu'
30
  le = LabelEncoder()
31
- le = joblib.load("/kaggle/working/SVD/le.gz")
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("/SVD/GeoG.pth", map_location=torch.device(device))
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'])