Spaces:

suko
/

nsfw

Running

suko commited on Mar 15, 2023

Commit

ca8afc0

•

1 Parent(s): ec8aa40

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ class ONNXModel:
         sorted_output = {"predictions": sorted(output, key=lambda k: k["confidence"], reverse=True)}
         return sorted_output
 EXPORT_MODEL_VERSION=1
-model = ONNXModel(dir_path="../model.onnx")
 model.load()
 def predict(image):

         sorted_output = {"predictions": sorted(output, key=lambda k: k["confidence"], reverse=True)}
         return sorted_output
 EXPORT_MODEL_VERSION=1
+model = ONNXModel(dir_path="/model/model.onnx")
 model.load()
 def predict(image):