(no commit message)
This commit is contained in:
23
main.py
Normal file
23
main.py
Normal file
@@ -0,0 +1,23 @@
|
||||
from modaic import PrecompiledAgent, PrecompiledConfig
|
||||
|
||||
class PromptToSignatureConfig(PrecompiledConfig):
|
||||
lm: str = "gpt-4o"
|
||||
max_tokens: int = 1024
|
||||
|
||||
class PromptToSignatureAgent(PrecompiledAgent):
|
||||
config: PromptToSignatureConfig
|
||||
|
||||
def __init__(self, config: PromptToSignatureConfig, **kwargs):
|
||||
super().__init__(config, **kwargs)
|
||||
|
||||
def forward(self, prompt: str) -> str:
|
||||
return "hello world"
|
||||
|
||||
|
||||
agent = PromptToSignatureAgent(PromptToSignatureConfig())
|
||||
|
||||
def main():
|
||||
agent.push_to_hub("fadeleke/prompt-to-signature", with_code=True)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user