Hugging Face's logo

Spaces:
drdoggo
/
Medical_Image_Understanding_with_VLMs
Runtime error

App Files Community
Thedatababbler commited on
Commit
3cbf763
·
1 Parent(s): a5f5ad5

app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def mlm(image, text):
39
 
40
  _, predicted_index = torch.topk(predictions[0][0][masked_index], 1)#.item()
41
  predicted_index = predicted_index.detach().cpu().numpy()
42
- #print(predicted_index)
43
  for idx in predicted_index:
44
  predicted_tokens.append(tokenizer.convert_ids_to_tokens([idx])[0])
45
  # for i in range(1):
 
39
 
40
  _, predicted_index = torch.topk(predictions[0][0][masked_index], 1)#.item()
41
  predicted_index = predicted_index.detach().cpu().numpy()
42
+ print(predicted_index)
43
  for idx in predicted_index:
44
  predicted_tokens.append(tokenizer.convert_ids_to_tokens([idx])[0])
45
  # for i in range(1):