Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ weight_softmax = params[-2].data.cpu().numpy() # The weights connecting features
|
|
| 58 |
|
| 59 |
def run_visual_detection(image):
|
| 60 |
if image is None: return None, "Waiting for feed..."
|
| 61 |
-
results = yolo_model.predict(image, conf=0.
|
| 62 |
return results[0].plot(), f"Objects Detected: {len(results[0].boxes)}"
|
| 63 |
|
| 64 |
def run_thermal_scan(image):
|
|
|
|
| 58 |
|
| 59 |
def run_visual_detection(image):
|
| 60 |
if image is None: return None, "Waiting for feed..."
|
| 61 |
+
results = yolo_model.predict(image, conf=0.65)
|
| 62 |
return results[0].plot(), f"Objects Detected: {len(results[0].boxes)}"
|
| 63 |
|
| 64 |
def run_thermal_scan(image):
|