Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|