Skip to content

Commit 457550f

Browse files
feat(predictor): add default model URLs for YOLO and RAMP in predictor module
1 parent 7fd07bc commit 457550f

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

predictor/app.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,13 @@ async def predict(
9494

9595
print(f"It took {round(time.time()-start)} sec to extract polygons")
9696

97+
if gdf.crs and gdf.crs != "EPSG:4326":
98+
gdf = gdf.to_crs("EPSG:4326")
99+
elif not gdf.crs:
100+
# if not defined assume its 3857 because above 3857 is hardcoded
101+
gdf.set_crs("EPSG:3857", inplace=True)
102+
gdf = gdf.to_crs("EPSG:4326")
103+
97104
gdf["building"] = "yes"
98105
gdf["source"] = "fAIr"
99106

0 commit comments

Comments
 (0)