Spaces:
Sleeping
Sleeping
add tools
Browse files
services/model_handler.py
CHANGED
@@ -10,7 +10,7 @@ import time
|
|
10 |
import datetime
|
11 |
import os
|
12 |
|
13 |
-
MODEL_PATH = "google/flan-t5-
|
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 |
|