Spaces:
Runtime error
Runtime error
arg
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ def predict(img: np.ndarray) -> str:
|
|
| 44 |
#print(result)
|
| 45 |
|
| 46 |
result_fp = 'temp_result.jpg'
|
| 47 |
-
cv2.imwrite(result_fp,
|
| 48 |
return result_fp
|
| 49 |
|
| 50 |
|
|
|
|
| 44 |
#print(result)
|
| 45 |
|
| 46 |
result_fp = 'temp_result.jpg'
|
| 47 |
+
cv2.imwrite(result_fp, result.cvtColor(result, cv2.cv2.COLOR_BGR2RGB))
|
| 48 |
return result_fp
|
| 49 |
|
| 50 |
|