ChandraP12330 commited on
Commit
6ce7076
·
verified ·
1 Parent(s): 7a598bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -87,7 +87,8 @@ if image_url:
87
  logits_per_image = outputs.logits_per_image # this is the image-text similarity score
88
  probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
89
  raw_context= scene_labels[probs.argmax(-1)]
90
- context= 'the image is depicting scene of '+ raw_context
 
91
 
92
  ##BLIP
93
  caption = image_to_text(image_url, max_new_tokens=200)
 
87
  logits_per_image = outputs.logits_per_image # this is the image-text similarity score
88
  probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
89
  raw_context= scene_labels[probs.argmax(-1)]
90
+ #context= 'the image is depicting scene of '+ raw_context
91
+ context= 'the image is depicting scene of attack'
92
 
93
  ##BLIP
94
  caption = image_to_text(image_url, max_new_tokens=200)