Hugging Face's logo

Spaces:
akhaliq
/
animeganv2-onnx
Runtime error

App Files Community
Ahsen Khaliq commited on
Commit
9139725
•
1 Parent(s): 8424e72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ os.system("pip install gdown")
6
  os.system("gdown https://drive.google.com/uc?id=1riNxV1BWMAXfmWZ3LrQbEkvzV8f7lOCp")
7
 
8
  opts = onnxruntime.SessionOptions()
9
- opts.intra_op_num_threads = 8
10
  onnx_session = onnxruntime.InferenceSession("/home/user/app/face_paint_512_v2_0.onnx",sess_options=opts)
11
 
12
  input_name = onnx_session.get_inputs()[0].name
 
6
  os.system("gdown https://drive.google.com/uc?id=1riNxV1BWMAXfmWZ3LrQbEkvzV8f7lOCp")
7
 
8
  opts = onnxruntime.SessionOptions()
9
+ opts.intra_op_num_threads = 16
10
  onnx_session = onnxruntime.InferenceSession("/home/user/app/face_paint_512_v2_0.onnx",sess_options=opts)
11
 
12
  input_name = onnx_session.get_inputs()[0].name