Update README.md

README.md

@@ -11,7 +11,7 @@ tags:
 - fashion
 - search
 ---
-This model is fine-tuned version of microsoft/conditional-detr-resnet-50.
+This model is fine-tuned version of hustvl/yolos-tiny.
 
 You can find details of model in this github repo -> [fashion-visual-search](https://github.com/yainage90/fashion-visual-search)
 
@@ -44,7 +44,7 @@ with torch.no_grad():
     inputs = image_processor(images=[image], return_tensors="pt")
     outputs = model(**inputs.to(device))
     target_sizes = torch.tensor([[image.size[1], image.size[0]]])
-    results = image_processor.post_process_object_detection(outputs, threshold=0.4, target_sizes=target_sizes)[0]
+    results = image_processor.post_process_object_detection(outputs, threshold=0.85, target_sizes=target_sizes)[0]
 
     items = []
     for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):