add google big query library
Browse files- .gitignore +8 -0
- app.py +2 -2
.gitignore
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# add .venv, python cache files to .gitignore
|
2 |
+
.venv/
|
3 |
+
__pycache__/
|
4 |
+
|
5 |
+
# add compiled python files to .gitignore
|
6 |
+
*.pyc
|
7 |
+
crisis_prediction.db
|
8 |
+
setup_duckdb_crisis_prediction.py
|
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
from datasets import load_dataset
|
4 |
from typing import Union, Dict, List, Optional
|
5 |
import pandas as pd
|
6 |
-
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
7 |
from opentelemetry import trace
|
8 |
from opentelemetry.sdk.trace import TracerProvider
|
9 |
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
@@ -92,7 +92,7 @@ model = HfApiModel()
|
|
92 |
agent = CodeAgent(
|
93 |
tools=[],
|
94 |
model=model,
|
95 |
-
additional_authorized_imports=["gradio","pandas","datasets","numpy","Bio","rdflib","networkx","requests","execute_sql","duckdb","pyarrow"],
|
96 |
add_base_tools=True
|
97 |
)
|
98 |
|
|
|
3 |
from datasets import load_dataset
|
4 |
from typing import Union, Dict, List, Optional
|
5 |
import pandas as pd
|
6 |
+
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
7 |
from opentelemetry import trace
|
8 |
from opentelemetry.sdk.trace import TracerProvider
|
9 |
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
|
|
92 |
agent = CodeAgent(
|
93 |
tools=[],
|
94 |
model=model,
|
95 |
+
additional_authorized_imports=["pyvis","aiohttp","os","sys","gradio","pandas","datasets","numpy","Bio","rdflib","networkx","requests","execute_sql","duckdb","pyarrow","google-cloud-bigquery"],
|
96 |
add_base_tools=True
|
97 |
)
|
98 |
|