Hugging Face's logo

Spaces:
frugal-ai-challenge
/
image-Amiel
Sleeping

App Files Community
kiurtis commited on
Commit
11d44cc
·
1 Parent(s): 11790f5

use half quantisized model

Browse files
Files changed (1) hide show
  1. tasks/image.py +1 -1
tasks/image.py CHANGED
@@ -141,7 +141,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
141
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
142
  #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
143
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
144
- PATH_TO_MODEL = 'models/best_YOLOv11n_1280_full_batch_32.engine'
145
  INFERENCE_ENGINE_TYPE = 'pt'
146
  INPUT_SIZE = 1280
147
  N_TEST_BATCHES = 2
 
141
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
142
  #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
143
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
144
+ PATH_TO_MODEL = 'models/best_YOLOv11n_1280_half_batch_32.engine'
145
  INFERENCE_ENGINE_TYPE = 'pt'
146
  INPUT_SIZE = 1280
147
  N_TEST_BATCHES = 2