Update Align_myself.py

uploading files locally as well

Align_myself.py

@@ -1,21 +1,20 @@
-import requests
 import torch
 from PIL import Image
-from transformers import AlignProcessor, AlignModel
+import requests
 import gradio as gr
+from transformers import AlignProcessor, AlignModel
 
 processor = AlignProcessor.from_pretrained("kakaobrain/align-base")
 model = AlignModel.from_pretrained("kakaobrain/align-base")
 
-
-def get_image_alignment_probabilities(url, is_url):
+def get_image_alignment_probabilities(image, is_url):
     candidate_labels = ["advertisement", "not an advertisement"]
 
-    # Load image from URL
+    # Load image from URL or locally
     if is_url:
-        image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
+        image = Image.open(requests.get(image, stream=True).raw).convert("RGB")
     else:
-        image = Image.open(url).convert("RGB")
+        image = Image.open(image).convert("RGB")
 
     # Process inputs
     inputs = processor(text=candidate_labels, images=image, return_tensors="pt")
@@ -32,9 +31,7 @@ def get_image_alignment_probabilities(url, is_url):
 
     return {label: prob.item() for label, prob in zip(candidate_labels, probs[0])}
 
-
 iface = gr.Interface(fn=get_image_alignment_probabilities,
-                     inputs=["text", "checkbox"],
+                     inputs=[gr.inputs.Image(type="file", label="Upload Image"), "checkbox"],
                      outputs="label")
 iface.launch()
-