Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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)
|