MikailDuzenli commited on
Commit
d7b5656
1 Parent(s): abc7e19

adding image for no heatmap

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. no_heatmap.jpg +0 -0
app.py CHANGED
@@ -171,7 +171,7 @@ def infer(url, mp_text, hidx):
171
  encoded = processor.tokenizer(inferred_token)
172
  output = processor.decode(encoded.input_ids[0], skip_special_tokens=True)
173
  selected_token = ''
174
- result = None
175
 
176
  if hidx > 0 and hidx < len(encoded["input_ids"][0][:-1]):
177
  input_ids = torch.tensor(encoded.input_ids)
 
171
  encoded = processor.tokenizer(inferred_token)
172
  output = processor.decode(encoded.input_ids[0], skip_special_tokens=True)
173
  selected_token = ''
174
+ result = Image.open('no_heatmap.jpg')
175
 
176
  if hidx > 0 and hidx < len(encoded["input_ids"][0][:-1]):
177
  input_ids = torch.tensor(encoded.input_ids)
no_heatmap.jpg ADDED