Upload app.py with huggingface_hub
Browse files
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(
|
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')
|