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