Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ model = hub.Module(name='se_hrnet64_imagenet_ssld')
|
|
7 |
def inference(img):
|
8 |
result = model.predict([img])
|
9 |
print(result)
|
10 |
-
for key, value in result[0]:
|
11 |
result[0][key] = float(value)
|
12 |
return result[0]
|
13 |
|
|
|
7 |
def inference(img):
|
8 |
result = model.predict([img])
|
9 |
print(result)
|
10 |
+
for key, value in result[0].items():
|
11 |
result[0][key] = float(value)
|
12 |
return result[0]
|
13 |
|