Spaces:
Runtime error
Runtime error
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)
|