Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|