Spaces:

onnx
/

FCN

Runtime error

akhaliq HF staff commited on Mar 24, 2022

Commit

b55b1ea

•

1 Parent(s): a301be4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,8 @@ preprocess = transforms.Compose([
 # based on the build flags) when instantiating InferenceSession.
 # For example, if NVIDIA GPU is available and ORT Python package is built with CUDA, then call API as following:
 # onnxruntime.InferenceSession(path/to/model, providers=['CUDAExecutionProvider'])
-sess = rt.InferenceSession("../model/fcn-resnet101-11.onnx")
 outputs = sess.get_outputs()
 output_names = list(map(lambda output: output.name, outputs))

 # based on the build flags) when instantiating InferenceSession.
 # For example, if NVIDIA GPU is available and ORT Python package is built with CUDA, then call API as following:
 # onnxruntime.InferenceSession(path/to/model, providers=['CUDAExecutionProvider'])
+os.system("wget https://github.com/AK391/models/raw/main/vision/object_detection_segmentation/fcn/model/fcn-resnet101-11.onnx")
+sess = rt.InferenceSession("fcn-resnet101-11.onnx")
 outputs = sess.get_outputs()
 output_names = list(map(lambda output: output.name, outputs))