debug

app.py

@@ -35,7 +35,7 @@ def classify_image(input_image) -> str:
         text=imagenet_classes,
         images=input_image,
         return_tensors="pt",
-        padding=True)
+        padding=True).to(device)
     outputs = model(**inputs)
     probs = outputs.logits_per_image.softmax(dim=1)
     class_index = np.argmax(probs.detach().numpy())