Vaibhav84 commited on
Commit
7dbf682
·
1 Parent(s): 06bd2d3

Db Connection added

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -3,9 +3,17 @@ import psycopg2
3
  import json
4
  from psycopg2 import sql
5
  import pandas as pd
 
 
 
 
 
 
 
 
6
  from DbConnection import DbConnection
7
  from UploadFile import UploadOpenFile
8
- from pydantic import BaseModel
9
  app = FastAPI()
10
 
11
  class FileDetails(BaseModel):
 
3
  import json
4
  from psycopg2 import sql
5
  import pandas as pd
6
+ import numpy as np
7
+ from pydantic import BaseModel
8
+ from psycopg2.extensions import register_adapter, AsIs
9
+ register_adapter(np.int64, AsIs)
10
+ import warnings
11
+ warnings.filterwarnings('ignore')
12
+
13
+ #Custom Classes for endpoints
14
  from DbConnection import DbConnection
15
  from UploadFile import UploadOpenFile
16
+
17
  app = FastAPI()
18
 
19
  class FileDetails(BaseModel):