drkareemkamal commited on
Commit
eb6f2d9
·
verified ·
1 Parent(s): 920e90c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -47,8 +47,10 @@ embedding_model = HuggingFaceEmbeddings(
47
  llm = HuggingFaceHub(
48
  # repo_id="epfl-llm/meditron-7b", # Or BioGPT, GatorTron, ClinicalT5, etc.
49
  # model_kwargs={"temperature": 0.4, "max_new_tokens": 512},
50
- repo_id="microsoft/BioGPT-Large",
51
- model_kwargs={"temperature": 0.4, "max_new_tokens": 512},
 
 
52
  huggingfacehub_api_token=hf_token
53
  )
54
 
 
47
  llm = HuggingFaceHub(
48
  # repo_id="epfl-llm/meditron-7b", # Or BioGPT, GatorTron, ClinicalT5, etc.
49
  # model_kwargs={"temperature": 0.4, "max_new_tokens": 512},
50
+ # repo_id="microsoft/BioGPT-Large",
51
+ # model_kwargs={"temperature": 0.4, "max_new_tokens": 512},
52
+ repo_id="emilyalsentzer/Bio_ClinicalBERT", # Encoder-only, fast
53
+ model_kwargs={"temperature": 0.3, "max_new_tokens": 256},
54
  huggingfacehub_api_token=hf_token
55
  )
56