diff --git a/main.py b/main.py index 66bd245..8e4feba 100644 --- a/main.py +++ b/main.py @@ -20,7 +20,7 @@ class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module() self.extract.set_lm(lm) def forward(self, image_url: str) -> str: - receipt_image = dspy.Image.from_url(image_url) + receipt_image = dspy.Image.from_url(url=image_url) return self.extract(receipt_image)