HugoVoxx commited on
Commit
dee2744
·
verified ·
1 Parent(s): fb4253b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -4
main.py CHANGED
@@ -7,10 +7,6 @@ if __name__ == '__main__':
7
  parser.add_argument('--model', type=str, default='gpt-4o-mini', help='Model to use for generation', choices=['gpt-4o-mini', 'gemma2-9b-it'])
8
  parser.add_argument('--question', type=str, required=True, help='Question to generate answer for')
9
  args = parser.parse_args()
10
- if args.model == 'gpt-4o-mini':
11
- get_github_token()
12
- elif args.model == 'gemma2-9b-it':
13
- get_grog_api()
14
  reader = Reader(model=args.model)
15
  input = translate(args.question)
16
  reader.main(input)
 
7
  parser.add_argument('--model', type=str, default='gpt-4o-mini', help='Model to use for generation', choices=['gpt-4o-mini', 'gemma2-9b-it'])
8
  parser.add_argument('--question', type=str, required=True, help='Question to generate answer for')
9
  args = parser.parse_args()
 
 
 
 
10
  reader = Reader(model=args.model)
11
  input = translate(args.question)
12
  reader.main(input)