Spaces:

mol-wise
/

MOLx-Powered_by_MedRAX

Paused

vaibhavm29 commited on May 23

Commit

5f09e75

1 Parent(s): 99c1e69

bug fixes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ def initialize_agent(
         "ChestXRaySegmentationTool": lambda: ChestXRaySegmentationTool(device=device),
         "LlavaMedTool": lambda: LlavaMedTool(cache_dir=model_dir, device=device, load_in_8bit=True),
         "XRayVQATool": lambda: XRayVQATool(cache_dir=model_dir, device=device),
-        "MedgemmaXRayTool": lambda: MedGemmaXRayTool(cache_dir=model_dir, device=device),
         "ChestXRayReportGeneratorTool": lambda: ChestXRayReportGeneratorTool(
             cache_dir=model_dir, device=device
         ),
@@ -107,7 +107,7 @@ if __name__ == "__main__":
         "ChestXRayReportGeneratorTool",
         "XRayVQATool",
         "LlavaMedTool",
-        "XRayPhraseGroundingTool",
         "MedGemmaXRayTool"
         # "ChestXRayGeneratorTool",
     ]

         "ChestXRaySegmentationTool": lambda: ChestXRaySegmentationTool(device=device),
         "LlavaMedTool": lambda: LlavaMedTool(cache_dir=model_dir, device=device, load_in_8bit=True),
         "XRayVQATool": lambda: XRayVQATool(cache_dir=model_dir, device=device),
+        "MedGemmaXRayTool": lambda: MedGemmaXRayTool(cache_dir=model_dir, device=device),
         "ChestXRayReportGeneratorTool": lambda: ChestXRayReportGeneratorTool(
             cache_dir=model_dir, device=device
         ),
         "ChestXRayReportGeneratorTool",
         "XRayVQATool",
         "LlavaMedTool",
+        # "XRayPhraseGroundingTool",
         "MedGemmaXRayTool"
         # "ChestXRayGeneratorTool",
     ]