Hugging Face's logo

Spaces:
Bman21
/
Medstudyeasyai
Sleeping

App Files Community
1
Bman21 commited on
Commit
09aa171
·
verified ·
1 Parent(s): f9e7c04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ SYSTEM_MESSAGE = (
21
  # MODELS
22
  # -------------------
23
  # PubMedBERT for embeddings
24
- EMBED_MODEL_NAME = "microsoft/BiomedNLP-PubMedBERT-base-uncased"
25
  embed_tokenizer = AutoTokenizer.from_pretrained(EMBED_MODEL_NAME)
26
  embed_model = AutoModel.from_pretrained(EMBED_MODEL_NAME)
27
 
 
21
  # MODELS
22
  # -------------------
23
  # PubMedBERT for embeddings
24
+ EMBED_MODEL_NAME = "microsoft/BiomedCLIP-PubMedBERT_256-vit_base_patch16_224"
25
  embed_tokenizer = AutoTokenizer.from_pretrained(EMBED_MODEL_NAME)
26
  embed_model = AutoModel.from_pretrained(EMBED_MODEL_NAME)
27