Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,13 +45,13 @@ embedding_model = HuggingFaceEmbeddings(
|
|
| 45 |
|
| 46 |
|
| 47 |
llm = HuggingFaceHub(
|
| 48 |
-
|
| 49 |
-
|
| 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 ====
|