Spaces:
Runtime error
Runtime error
input details index get back
Browse files- extractor.py +1 -1
extractor.py
CHANGED
|
@@ -28,7 +28,7 @@ def get_card_xy(model_path, image_path):
|
|
| 28 |
image_np = image_np[np.newaxis, :]
|
| 29 |
|
| 30 |
# inference
|
| 31 |
-
interpreter.set_tensor(
|
| 32 |
interpreter.invoke()
|
| 33 |
|
| 34 |
# Obtaining output results
|
|
|
|
| 28 |
image_np = image_np[np.newaxis, :]
|
| 29 |
|
| 30 |
# inference
|
| 31 |
+
interpreter.set_tensor(input_details[0]['index'], image_np)
|
| 32 |
interpreter.invoke()
|
| 33 |
|
| 34 |
# Obtaining output results
|