Update app.py

app.py

@@ -64,7 +64,7 @@ def process_image(image):
 
     # Predict token labels
     with torch.no_grad():
-        outputs = model.predict_token_labels(input_ids, bbox, attention_mask=attention_mask)
+        outputs = model(input_ids=input_ids, bbox=bbox, attention_mask=attention_mask)
 
     predicted_labels = outputs.predictions.argmax(dim=2).squeeze().tolist()