(no commit message)
This commit is contained in:
18
main.py
18
main.py
@@ -1,15 +1,13 @@
|
||||
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
|
||||
): # 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()
|
||||
class ReceiptClassifier(PrecompiledAgent): # Extends dspy.module()
|
||||
config: ReceiptClassifierConfig
|
||||
|
||||
def __init__(self, config: ReceiptClassifierConfig, **kwargs):
|
||||
@@ -23,13 +21,5 @@ class ReceiptClassifier(PrecompiledAgent): # extends dspy.module()
|
||||
receipt_image = dspy.Image(url=image_url)
|
||||
return self.extract(image=receipt_image)
|
||||
|
||||
|
||||
agent = ReceiptClassifier(ReceiptClassifierConfig())
|
||||
|
||||
|
||||
def main():
|
||||
agent.push_to_hub("farouk1/receipt-classifier", with_code=True)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
agent.push_to_hub("farouk1/receipt-classifier", with_code=True) # Set MODAIC_TOKEN
|
||||
|
||||
Reference in New Issue
Block a user