Update app.py
Browse files
app.py
CHANGED
|
@@ -9,15 +9,19 @@ model = AutoModel.from_pretrained('jinaai/jina-clip-v1', trust_remote_code=True)
|
|
| 9 |
def compute_similarity(input1_type, input1_text, input1_image, input2_type, input2_text, input2_image):
|
| 10 |
"""Computes similarity for Text-Text, Image-Image, or Text-Image comparisons."""
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
# Handle empty inputs properly
|
| 13 |
if input1_type == "Text" and not input1_text.strip():
|
| 14 |
return "Error: Input 1 (Text) is empty!"
|
| 15 |
-
if input1_type == "Image" and
|
| 16 |
return "Error: Please upload a valid image for Input 1!"
|
| 17 |
|
| 18 |
if input2_type == "Text" and not input2_text.strip():
|
| 19 |
return "Error: Input 2 (Text) is empty!"
|
| 20 |
-
if input2_type == "Image" and
|
| 21 |
return "Error: Please upload a valid image for Input 2!"
|
| 22 |
|
| 23 |
try:
|
|
|
|
| 9 |
def compute_similarity(input1_type, input1_text, input1_image, input2_type, input2_text, input2_image):
|
| 10 |
"""Computes similarity for Text-Text, Image-Image, or Text-Image comparisons."""
|
| 11 |
|
| 12 |
+
# Ensure images are valid (Gradio may pass `False` instead of `None`)
|
| 13 |
+
input1_image = None if isinstance(input1_image, bool) else input1_image
|
| 14 |
+
input2_image = None if isinstance(input2_image, bool) else input2_image
|
| 15 |
+
|
| 16 |
# Handle empty inputs properly
|
| 17 |
if input1_type == "Text" and not input1_text.strip():
|
| 18 |
return "Error: Input 1 (Text) is empty!"
|
| 19 |
+
if input1_type == "Image" and input1_image is None:
|
| 20 |
return "Error: Please upload a valid image for Input 1!"
|
| 21 |
|
| 22 |
if input2_type == "Text" and not input2_text.strip():
|
| 23 |
return "Error: Input 2 (Text) is empty!"
|
| 24 |
+
if input2_type == "Image" and input2_image is None:
|
| 25 |
return "Error: Please upload a valid image for Input 2!"
|
| 26 |
|
| 27 |
try:
|