Hugging Face's logo

Spaces:
torchFlow
/
detect-it
Sleeping

App Files Community
Arnav Raina commited on
Commit
2c597e4
·
1 Parent(s): 282d8f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,10 +5,10 @@ PRED = []
5
 
6
  def detect(files):
7
  model = ONNXInference(
8
- model_path="/home/neo/Downloads/torchFlow/models/torchFlow-ckpt.onnx",
9
  files=files,
10
  save_image=False,
11
- save_path="/home/neo/Downloads/torchFlow/"
12
  )
13
  res = model.run()
14
  img_id = res["IMG_ID"]
 
5
 
6
  def detect(files):
7
  model = ONNXInference(
8
+ model_path="./torchFlow-ckpt.onnx",
9
  files=files,
10
  save_image=False,
11
+ save_path="./"
12
  )
13
  res = model.run()
14
  img_id = res["IMG_ID"]