File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -60,8 +60,8 @@ async def predict(
60
60
61
61
os .makedirs (base_path , exist_ok = True )
62
62
meta_path , results_path = (
63
- os .path .join (output_path , "meta" ),
64
- os .path .join (output_path , "results" ),
63
+ os .path .join (base_path , "meta" ),
64
+ os .path .join (base_path , "results" ),
65
65
)
66
66
os .makedirs (meta_path , exist_ok = True )
67
67
os .makedirs (results_path , exist_ok = True )
Original file line number Diff line number Diff line change @@ -91,6 +91,13 @@ class PredictionRequest(BaseModel):
91
91
default = False ,
92
92
description = "Whether to include predictions as points, this will create output geojson with extra points predictions" ,
93
93
)
94
+ remove_metadata : Optional [bool ] = Field (
95
+ default = True ,
96
+ description = "Whether to remove intermediate metadata files after processing" ,
97
+ )
98
+ output_path : Optional [str ] = Field (
99
+ default = None , description = "Path to save the output files"
100
+ )
94
101
95
102
@field_validator ("checkpoint" )
96
103
def validate_checkpoint (cls , value ):
You can’t perform that action at this time.
0 commit comments