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 |
|