Update app.py

app.py

@@ -36,13 +36,7 @@ st.title("Image Caption Surveillance")
 
 # Input field for URL
 image_url = st.text_input("Enter the URL of the image:")
-image = Image.open(requests.get(image_url, stream=True).raw)
-inputs = processor(text=scene_labels, images=image, return_tensors="pt", padding=True)
-outputs = model(**inputs)
-logits_per_image = outputs.logits_per_image # this is the image-text similarity score
-probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
 
-context= scene_labels[probs.argmax(-1)]
 
 
 
@@ -50,6 +44,13 @@ context= scene_labels[probs.argmax(-1)]
 if image_url:
     try:
         st.image(image_url, caption="Uploaded Image")
+        image = Image.open(requests.get(image_url, stream=True).raw)
+        inputs = processor(text=scene_labels, images=image, return_tensors="pt", padding=True)
+        outputs = model(**inputs)
+        logits_per_image = outputs.logits_per_image # this is the image-text similarity score
+        probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
+
+        context= scene_labels[probs.argmax(-1)]
         st.write("context: ", context)
     except Exception as e:
         st.error(f"Error: {e}")