Update app.py

app.py

@@ -4,37 +4,13 @@ from io import BytesIO
 import gradio as gr
 
 
-# Initialize CLIP model and processor
-processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
-model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
-
-def image_similarity(image: Image.Image, action_prompt: str) -> bool:
-    positive_text = f"a picture of someone {action_prompt}"
-    negative_text = f"other"
-
-    inputs = processor(
-        text=[positive_text, negative_text], 
-        images=image, 
-        return_tensors="pt", 
-        padding=True
-    )
-
-    outputs = model(**inputs)
-    logits_per_image = outputs.logits_per_image  # image-text similarity score
-    probs = logits_per_image.softmax(dim=1)  # take the softmax to get the label probabilities
-    
-    # Determine if positive prompt has a higher probability than the negative prompt
-    result = probs[0][0] > probs[0][1]
-    return result
-
-
 # Initialize CLIP model and processor
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
 
 def image_similarity(image: Image.Image, action_prompt: str):
-    positive_text = f"a picture of someone {action_prompt}"
-    negative_text = f"not a picture of someone {action_prompt}"
+    positive_text = f"a picture of a person {action_prompt}"
+    negative_text = f"not a picture a person {action_prompt}"
 
     inputs = processor(
         text=[positive_text, negative_text],