akhaliq HF staff commited on
Commit
5dee410
1 Parent(s): 18c75e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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