Spaces:
Running
Running
File size: 681 Bytes
78dd5ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from reader import Reader
from utils import *
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Generate Geometry Answer')
parser.add_argument('--model', type=str, default='gpt-4o-mini', help='Model to use for generation', choices=['gpt-4o-mini', 'gemma2-9b-it'])
parser.add_argument('--question', type=str, required=True, help='Question to generate answer for')
args = parser.parse_args()
if args.model == 'gpt-4o-mini':
get_github_token()
elif args.model == 'gemma2-9b-it':
get_grog_api()
reader = Reader(model=args.model)
input = translate(args.question)
reader.main(input)
run_ag() |