Spaces:
Runtime error
Runtime error
Nadine Rueegg
commited on
Commit
•
cc1d524
1
Parent(s):
4c1801d
small bug fix
Browse files
gradio_demo/barc_demo_v6.py
CHANGED
|
@@ -37,7 +37,7 @@ print(
|
|
| 37 |
"\ntorchvision: ", torchvision.__version__,
|
| 38 |
)
|
| 39 |
|
| 40 |
-
|
| 41 |
|
| 42 |
def get_prediction(model, img_path_or_img, confidence=0.5):
|
| 43 |
"""
|
|
@@ -191,7 +191,7 @@ def run_barc_inference(input_image, bbox=None):
|
|
| 191 |
return result_gltf
|
| 192 |
# -------------------------------------------------------------------------------------------------------------------- #
|
| 193 |
|
| 194 |
-
|
| 195 |
def run_complete_inference(img_path_or_img, crop_choice):
|
| 196 |
total_count += 1
|
| 197 |
print('total count: ' + str(total_count))
|
|
|
|
| 37 |
"\ntorchvision: ", torchvision.__version__,
|
| 38 |
)
|
| 39 |
|
| 40 |
+
|
| 41 |
|
| 42 |
def get_prediction(model, img_path_or_img, confidence=0.5):
|
| 43 |
"""
|
|
|
|
| 191 |
return result_gltf
|
| 192 |
# -------------------------------------------------------------------------------------------------------------------- #
|
| 193 |
|
| 194 |
+
total_count = 0
|
| 195 |
def run_complete_inference(img_path_or_img, crop_choice):
|
| 196 |
total_count += 1
|
| 197 |
print('total count: ' + str(total_count))
|