[CLEANUP]

kyegomez · Feb 17, 2024 · 7ebf605 · 7ebf605
1 parent c541714
commit 7ebf605
Show file tree

Hide file tree

Showing 5 changed files with 930 additions and 716 deletions.
diff --git a/example.py b/example.py
@@ -1,12 +1,10 @@
 import torch
 from medpalm.model import MedPalm
 
-#usage
+# usage
 img = torch.randn(1, 3, 256, 256)
 caption = torch.randint(0, 20000, (1, 1024))
 
 model = MedPalm()
 output = model(img, caption)
-print(output.shape) # (1, 1024, 20000)
-
-
+print(output.shape)  # (1, 1024, 20000)
diff --git a/medpalm/__init__.py b/medpalm/__init__.py
@@ -1 +1,3 @@
-from medpalm.model import MedPalm
+from medpalm.model import MedPalm
+
+__all__ = ["MedPalm"]