Spaces:
Runtime error
Runtime error
Chauhan
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,14 +43,14 @@ def main():
|
|
| 43 |
rag_subparser = rag_parser.add_subparsers(title='subcommand', dest='subcommand', description='Choose a sub command to start the app.')
|
| 44 |
rag_start_parser = rag_subparser.add_parser("start", help="Start the rag app")
|
| 45 |
rag_start_parser.add_argument('--assistant-type', type=str,
|
| 46 |
-
choices=["healthcare"],
|
| 47 |
default="default",
|
| 48 |
help='Assistant Type')
|
| 49 |
|
| 50 |
args = parser.parse_args()
|
| 51 |
args.command = 'rag'
|
| 52 |
args.subcommand = 'start'
|
| 53 |
-
args.assistant_type = '
|
| 54 |
|
| 55 |
if args.command == 'rag' and args.subcommand == 'start':
|
| 56 |
_start_vuln_rag(args)
|
|
|
|
| 43 |
rag_subparser = rag_parser.add_subparsers(title='subcommand', dest='subcommand', description='Choose a sub command to start the app.')
|
| 44 |
rag_start_parser = rag_subparser.add_parser("start", help="Start the rag app")
|
| 45 |
rag_start_parser.add_argument('--assistant-type', type=str,
|
| 46 |
+
choices=["healthcare", "diabetes"],
|
| 47 |
default="default",
|
| 48 |
help='Assistant Type')
|
| 49 |
|
| 50 |
args = parser.parse_args()
|
| 51 |
args.command = 'rag'
|
| 52 |
args.subcommand = 'start'
|
| 53 |
+
args.assistant_type = 'healthcare'
|
| 54 |
|
| 55 |
if args.command == 'rag' and args.subcommand == 'start':
|
| 56 |
_start_vuln_rag(args)
|