mdanish commited on
Commit
9cefacb
·
verified ·
1 Parent(s): a312e49

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,7 +28,7 @@ st.write("Available models:", open_clip.list_models())
28
  def load_model():
29
  """Load the OpenCLIP model and return model and processor"""
30
  model, _, preprocess = open_clip.create_model_and_transforms(
31
- 'ViT-H-14',
32
  pretrained='laion2b_s32b_b79k'
33
  )
34
  tokenizer = open_clip.get_tokenizer('ViT-H-14')
@@ -57,7 +57,7 @@ def main():
57
  st.error(f"Error loading model: {str(e)}")
58
  st.info("Please make sure you have enough memory and the correct dependencies installed.")
59
 
60
- knn = np.load(modelpath)
61
  st.write(knn['walkability_vecs'].shape)
62
 
63
  file = st.file_uploader('Upload An Image')
 
28
  def load_model():
29
  """Load the OpenCLIP model and return model and processor"""
30
  model, _, preprocess = open_clip.create_model_and_transforms(
31
+ 'ViT-H-14-378-quickgelu',
32
  pretrained='laion2b_s32b_b79k'
33
  )
34
  tokenizer = open_clip.get_tokenizer('ViT-H-14')
 
57
  st.error(f"Error loading model: {str(e)}")
58
  st.info("Please make sure you have enough memory and the correct dependencies installed.")
59
 
60
+ knn = np.load(knnpath)
61
  st.write(knn['walkability_vecs'].shape)
62
 
63
  file = st.file_uploader('Upload An Image')