rvv-karma commited on
Commit
34cfea2
·
verified ·
1 Parent(s): 1f098f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from PIL import Image
7
  from transformers import pipeline
8
  pipe = pipeline("image-classification", "rvv-karma/Human-Action-Recognition-VIT-Base-patch16-224")
9
 
10
- def classify_image(input):
11
- image = Image.fromarray(input.astype('uint8'), 'RGB')
12
  predictions = pipe(image)
13
  return {prediction["label"]: prediction["score"] for prediction in predictions}
14
 
 
7
  from transformers import pipeline
8
  pipe = pipeline("image-classification", "rvv-karma/Human-Action-Recognition-VIT-Base-patch16-224")
9
 
10
+ def classify_image(input_data):
11
+ image = Image.fromarray(input_data.astype('uint8'), 'RGB')
12
  predictions = pipe(image)
13
  return {prediction["label"]: prediction["score"] for prediction in predictions}
14