Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -264,14 +264,14 @@ async def download_file(fileId: str):
|
|
264 |
return {"error": "File not found"}
|
265 |
|
266 |
def load_onnx(model, sess_options, providers = ["CPUExecutionProvider"]):
|
267 |
-
|
268 |
config = load_config(model)
|
269 |
model = onnxruntime.InferenceSession(
|
270 |
str(model),
|
271 |
sess_options=sess_options,
|
272 |
providers= providers
|
273 |
)
|
274 |
-
|
275 |
return model, config
|
276 |
|
277 |
def load_config(model):
|
|
|
264 |
return {"error": "File not found"}
|
265 |
|
266 |
def load_onnx(model, sess_options, providers = ["CPUExecutionProvider"]):
|
267 |
+
# _LOGGER.debug("Loading model from %s", model)
|
268 |
config = load_config(model)
|
269 |
model = onnxruntime.InferenceSession(
|
270 |
str(model),
|
271 |
sess_options=sess_options,
|
272 |
providers= providers
|
273 |
)
|
274 |
+
# _LOGGER.info("Loaded model from %s", model)
|
275 |
return model, config
|
276 |
|
277 |
def load_config(model):
|