Spaces:
Sleeping
Sleeping
Edit conf value
Browse files
main.py
CHANGED
|
@@ -47,7 +47,7 @@ async def predict(request: ImageRequest):
|
|
| 47 |
if image is None:
|
| 48 |
return {"error": "Invalid image"}
|
| 49 |
|
| 50 |
-
results = model.predict(image)
|
| 51 |
result = results[0]
|
| 52 |
|
| 53 |
json_result = {}
|
|
|
|
| 47 |
if image is None:
|
| 48 |
return {"error": "Invalid image"}
|
| 49 |
|
| 50 |
+
results = model.predict(image, conf=0.36)
|
| 51 |
result = results[0]
|
| 52 |
|
| 53 |
json_result = {}
|