Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def plot_results(image, results, threshold=0.7):
|
|
25 |
def predict(image):
|
26 |
# make the object detection pipeline
|
27 |
obj_detector = pipeline(
|
28 |
-
"object-detection", model="Antoine101/detr-resnet-50-dc5-fashionpedia-finetuned"
|
29 |
)
|
30 |
results = obj_detector(image)
|
31 |
print("Results:")
|
|
|
25 |
def predict(image):
|
26 |
# make the object detection pipeline
|
27 |
obj_detector = pipeline(
|
28 |
+
"object-detection", model="Antoine101/detr-resnet-50-dc5-fashionpedia-finetuned-10percent-5000steps"
|
29 |
)
|
30 |
results = obj_detector(image)
|
31 |
print("Results:")
|