Akjava commited on
Commit
1c500a7
·
1 Parent(s): ff3c171

remove onnx

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -372,6 +372,8 @@ def find_angles(image,order):
372
  y_pred = model[i].predict(x)
373
  result_preds.append(y_pred.round(2))
374
  return result_preds # yxz-orderd x,y,z
 
 
375
  import onnxruntime as ort
376
  def estimate3(model_key,features_values):
377
  model_path=f"models/{model_key}.onnx"
@@ -385,6 +387,7 @@ def find_angles(image,order):
385
  result_preds = ort_session.run(None, {input_name: input_data})
386
  #print((result_preds))
387
  return result_preds[0] # yxz-orderd x,y,z
 
388
 
389
 
390
  #short_result = estimate('linear-svr-xyz_5.joblib','linear-svr-xyz_5_scaler.joblib',features_values)
 
372
  y_pred = model[i].predict(x)
373
  result_preds.append(y_pred.round(2))
374
  return result_preds # yxz-orderd x,y,z
375
+
376
+ """
377
  import onnxruntime as ort
378
  def estimate3(model_key,features_values):
379
  model_path=f"models/{model_key}.onnx"
 
387
  result_preds = ort_session.run(None, {input_name: input_data})
388
  #print((result_preds))
389
  return result_preds[0] # yxz-orderd x,y,z
390
+ """
391
 
392
 
393
  #short_result = estimate('linear-svr-xyz_5.joblib','linear-svr-xyz_5_scaler.joblib',features_values)