dwb2023 commited on
Commit
b78d488
·
1 Parent(s): 8879b88

add google big query library

Browse files
Files changed (2) hide show
  1. .gitignore +8 -0
  2. 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, ToolCallingAgent
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