safinal commited on
Commit
b0bd1f4
Β·
verified Β·
1 Parent(s): e3ae1d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ def encode_database(model, df: pd.DataFrame) -> np.ndarray :
40
 
41
  # Load model and configurations
42
  def load_model():
43
- model = Model(model_name="ViTamin-L-384", pretrained=None)
44
- model.load("weights.pth")
45
  model.eval()
46
  return model
47
 
 
40
 
41
  # Load model and configurations
42
  def load_model():
43
+ model = Model(model_name="ViTamin-L-384", pretrained="datacomp1b")
44
+ # model.load("weights.pth")
45
  model.eval()
46
  return model
47