Spaces:
Runtime error
Runtime error
Commit
·
983d2ca
1
Parent(s):
fa8c93c
changes in xray_vqa tool
Browse files- medrax/tools/xray_vqa.py +2 -1
medrax/tools/xray_vqa.py
CHANGED
|
@@ -45,7 +45,8 @@ class XRayVQATool(BaseTool):
|
|
| 45 |
|
| 46 |
def __init__(
|
| 47 |
self,
|
| 48 |
-
model_name: str = "StanfordAIMI/CheXagent-2-3b",
|
|
|
|
| 49 |
device: Optional[str] = "cuda",
|
| 50 |
dtype: torch.dtype = torch.bfloat16,
|
| 51 |
cache_dir: Optional[str] = None,
|
|
|
|
| 45 |
|
| 46 |
def __init__(
|
| 47 |
self,
|
| 48 |
+
# model_name: str = "StanfordAIMI/CheXagent-2-3b",
|
| 49 |
+
model_name: str = "google/medgemma-4b-it",
|
| 50 |
device: Optional[str] = "cuda",
|
| 51 |
dtype: torch.dtype = torch.bfloat16,
|
| 52 |
cache_dir: Optional[str] = None,
|