Spaces:
Sleeping
Sleeping
Sergey Kolbin
commited on
Commit
·
28dc509
1
Parent(s):
8aa9d72
m1
Browse files- app.py +4 -4
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -97,10 +97,10 @@ def count_big_cats(img, score_threshold, iou_threshold):
|
|
| 97 |
return tiger_count, lion_count, total_count, img_annotated
|
| 98 |
|
| 99 |
TEST_IMAGES = {
|
| 100 |
-
"
|
| 101 |
-
"
|
| 102 |
-
"
|
| 103 |
-
"
|
| 104 |
}
|
| 105 |
|
| 106 |
def load_test_image(choice):
|
|
|
|
| 97 |
return tiger_count, lion_count, total_count, img_annotated
|
| 98 |
|
| 99 |
TEST_IMAGES = {
|
| 100 |
+
"Tigers": "examples/tiger1.png",
|
| 101 |
+
"More Tigers": "examples/tiger2.png",
|
| 102 |
+
"Funny Tigers": "examples/tiger3.png",
|
| 103 |
+
"Lions": "examples/tigers_and_lions_2.png",
|
| 104 |
}
|
| 105 |
|
| 106 |
def load_test_image(choice):
|
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
transformers>=4.41.0
|
| 2 |
huggingface_hub>=0.23.0
|
| 3 |
torch
|
|
|
|
| 4 |
gradio>=4.0.0
|
| 5 |
pillow
|
| 6 |
safetensors
|
|
|
|
| 1 |
transformers>=4.41.0
|
| 2 |
huggingface_hub>=0.23.0
|
| 3 |
torch
|
| 4 |
+
scipy
|
| 5 |
gradio>=4.0.0
|
| 6 |
pillow
|
| 7 |
safetensors
|