jit fix 4

app.py

@@ -50,7 +50,7 @@ def process_image(image):
 
     # Extract the features from the uploaded image
     with torch.no_grad():
-        inputs = processor(images=image, return_tensors="pt").to(device)
+        inputs = processor(images=image, return_tensors="pt")["pixel_values"].to(device)
 
         # Use the traced model for inference
         outputs = traced_model(**inputs)