Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ transform = Compose([
|
|
29 |
def classify_image(image):
|
30 |
# Preprocess the image
|
31 |
image = Image.fromarray(image)
|
32 |
-
inputs = processor(text=
|
33 |
|
34 |
# Run inference
|
35 |
outputs = model(**inputs)
|
|
|
29 |
def classify_image(image):
|
30 |
# Preprocess the image
|
31 |
image = Image.fromarray(image)
|
32 |
+
inputs = processor(text=labels, images=image, return_tensors="pt", padding=True)
|
33 |
|
34 |
# Run inference
|
35 |
outputs = model(**inputs)
|