akhaliq HF staff commited on
Commit
b365624
1 Parent(s): 518bee4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def inference(img):
8
  input_dict = {"image": [test_img_path]}
9
  result = classifier.classification(data=input_dict)
10
  print(result)
11
- return result[0]
12
 
13
 
14
  title="resnet_v2_34_imagenet"
8
  input_dict = {"image": [test_img_path]}
9
  result = classifier.classification(data=input_dict)
10
  print(result)
11
+ return result[0][0]
12
 
13
 
14
  title="resnet_v2_34_imagenet"