wakeupmh commited on
Commit
0499f49
·
1 Parent(s): 9d8a332
Files changed (1) hide show
  1. services/model_handler.py +5 -1
services/model_handler.py CHANGED
@@ -10,7 +10,7 @@ import time
10
  import datetime
11
  import os
12
 
13
- MODEL_PATH = "google/flan-t5-small"
14
 
15
  # Simple Response class to wrap the model output
16
  class Response:
@@ -454,6 +454,10 @@ Output:"""
454
  "The content must be for people without autism knowledge.",
455
  "Focus on providing comprehensive information about the topic.",
456
  "Include definition, characteristics, causes, and current understanding."
 
 
 
 
457
  ]
458
  )
459
 
 
10
  import datetime
11
  import os
12
 
13
+ MODEL_PATH = "google/flan-t5-base"
14
 
15
  # Simple Response class to wrap the model output
16
  class Response:
 
454
  "The content must be for people without autism knowledge.",
455
  "Focus on providing comprehensive information about the topic.",
456
  "Include definition, characteristics, causes, and current understanding."
457
+ ],
458
+ tools=[
459
+ ArxivTools(),
460
+ PubmedTools()
461
  ]
462
  )
463