Ahsen Khaliq commited on
Commit
e6d701a
1 Parent(s): 431168d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  os.system("pip install gdown")
4
  os.system("gdown https://drive.google.com/uc?id=1riNxV1BWMAXfmWZ3LrQbEkvzV8f7lOCp")
5
 
6
- onnx_session = onnxruntime.InferenceSession("face_paint_512_v2_0.onnx")
7
 
8
  input_name = onnx_session.get_inputs()[0].name
9
  output_name = onnx_session.get_outputs()[0].name
 
3
  os.system("pip install gdown")
4
  os.system("gdown https://drive.google.com/uc?id=1riNxV1BWMAXfmWZ3LrQbEkvzV8f7lOCp")
5
 
6
+ onnx_session = onnxruntime.InferenceSession("/home/user/app/face_paint_512_v2_0.onnx")
7
 
8
  input_name = onnx_session.get_inputs()[0].name
9
  output_name = onnx_session.get_outputs()[0].name