| import argparse,asyncio,json |
| from pathlib import Path |
| import torch |
| from coreai_decoder import AmosDecoder |
| async def main(name): |
| def forbidden(*a,**kw):raise AssertionError('PyTorch neural execution forbidden') |
| torch.nn.Linear.forward=forbidden;torch.nn.Embedding.forward=forbidden;torch.nn.Conv2d.forward=forbidden;torch.nn.Conv3d.forward=forbidden |
| root=Path('bundles')/(name+'-CoreAI');m=await AmosDecoder.load(root) |
| out=await m.request({'state':'Please refund the duplicate charge.','questions':{'refund':{'type':'noul','instructions':'Is a refund requested?'}}}) |
| assert out['answers']['refund']['noul']>0.5,out |
| from PIL import Image |
| image_result=await m.predict('A red image.',[{'type':'region','text':'Point to the red area.'}],[Image.new('RGB',(360,240),'red')]) |
| assert image_result[0]['type']=='region' |
| try:await m.predict('word '*1000,[{'type':'noul','text':'Is there text?'}]) |
| except ValueError:capacity_guard=True |
| else:raise AssertionError('Over-capacity request was accepted') |
| (root/'runtime-smoke.json').write_text(json.dumps({'passed':True,'pytorch_neural_execution_disabled':True,'capacity_guard':capacity_guard,'result':out,'image_result':image_result},indent=2));print(out) |
| if __name__=='__main__': |
| p=argparse.ArgumentParser();p.add_argument('name');a=p.parse_args();asyncio.run(main(a.name)) |
|
|
