farouk1/receipt-classifier
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(no commit message)

Browse Source
This commit is contained in:
Farouk Adeleke
2025-10-25 02:39:01 -04:00
parent 4f901a2bbe
commit bafd41de2a
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -1,12 +1,14 @@
from modaic import PrecompiledAgent, PrecompiledConfig from modaic import PrecompiledAgent, PrecompiledConfig
import dspy import dspy
class ReceiptClassifierConfig( class ReceiptClassifierConfig(
PrecompiledConfig PrecompiledConfig
): # Configurable for each agent instance ): # Configurable for each agent instance
lm: str = "openai/gpt-4o-mini" # Set OPENAI_API_KEY lm: str = "openai/gpt-4o-mini" # Set OPENAI_API_KEY
max_tokens: int = 500 max_tokens: int = 500
class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module() class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module()
config: ReceiptClassifierConfig config: ReceiptClassifierConfig
@@ -21,11 +23,13 @@ class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module()
receipt_image = dspy.Image.from_url(image_url) receipt_image = dspy.Image.from_url(image_url)
return self.extract(receipt_image) return self.extract(receipt_image)
agent = ReceiptClassifier(ReceiptClassifierConfig()) agent = ReceiptClassifier(ReceiptClassifierConfig())
def main(): def main():
agent.push_to_hub("farouk1/receipt-classifier", with_code=True) agent.push_to_hub("farouk1/receipt-classifier", with_code=True)
if __name__ == "__main__": if __name__ == "__main__":
main() main()