Upload app.py with huggingface_hub

app.py

@@ -8,6 +8,7 @@ import open_clip
 
 knnpath = '20241204-ams-no-env-open_clip_ViT-H-14-378-quickgelu.npz'
 clip_model_name = 'ViT-H-14-378-quickgelu'
+pretrained_name = 'dfn5b'
 
 
 # Set page config
@@ -22,16 +23,15 @@ st.set_page_config(
 #model, preprocess = open_clip.create_model_from_pretrained(clip_model_name)
 #tokenizer = open_clip.get_tokenizer(clip_model_name)
 
-st.write("Available models:", open_clip.list_models())
+#st.write("Available models:", open_clip.list_models())
 
 @st.cache_resource
 def load_model():
     """Load the OpenCLIP model and return model and processor"""
     model, _, preprocess = open_clip.create_model_and_transforms(
-        'ViT-H-14-378-quickgelu',
-        pretrained='laion2b_s32b_b79k'
+        clip_model_name, pretrained=pretrained_name
     )
-    tokenizer = open_clip.get_tokenizer('ViT-H-14')
+    tokenizer = open_clip.get_tokenizer(clip_model_name)
     return model, preprocess, tokenizer
 
 def process_image(image, preprocess):