drkareemkamal commited on
Commit
52a148f
·
verified ·
1 Parent(s): eb6f2d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -45,13 +45,13 @@ embedding_model = HuggingFaceEmbeddings(
45
 
46
 
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
 
57
  # ==== Helpers ====
 
45
 
46
 
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=os.getenv('hf_token')
55
  )
56
 
57
  # ==== Helpers ====