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