(no commit message)
This commit is contained in:
2
main.py
2
main.py
@@ -20,7 +20,7 @@ class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module()
|
|||||||
self.extract.set_lm(lm)
|
self.extract.set_lm(lm)
|
||||||
|
|
||||||
def forward(self, image_url: str) -> str:
|
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)
|
return self.extract(receipt_image)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user