Hugging Face's logo

Spaces:
Norod78
/
PumpkinHeads
Sleeping

App Files Community
Norod78 commited on
Commit
f18cdf6
•
1 Parent(s): f016b20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ MODEL_FILE = "ffhq2pumpkinheads_pix2pixHD_e03_itr4500-simp.onnx"
14
  so = rt.SessionOptions()
15
  so.inter_op_num_threads = 4
16
  so.intra_op_num_threads = 4
17
- session = rt.InferenceSession(MODEL_FILE, sess_options=so)
18
  input_name = session.get_inputs()[0].name
19
  print("input_name = " + str(input_name))
20
  output_name = session.get_outputs()[0].name
 
14
  so = rt.SessionOptions()
15
  so.inter_op_num_threads = 4
16
  so.intra_op_num_threads = 4
17
+ session = rt.InferenceSession(MODEL_FILE, sess_options=so,providers=['CPUExecutionProvider'])
18
  input_name = session.get_inputs()[0].name
19
  print("input_name = " + str(input_name))
20
  output_name = session.get_outputs()[0].name