for testing purposes

testing_api.py

@@ -0,0 +1,44 @@
+import requests
+import json
+import time
+
+# API base URL (replace with your actual Hugging Face Space URL)
+API_URL = "https://fbetteo-clip-vit-b-32-test.hf.space"
+
+
+def get_image_embedding(image_url: str) -> dict:
+    """
+    Get image embedding from the CLIP API.
+
+    Args:
+        image_url: URL of the image to process
+
+    Returns:
+        Dictionary containing embedding and dimension
+    """
+    endpoint = f"{API_URL}/embed-image"
+    payload = {"image_url": image_url}
+
+    response = requests.post(endpoint, json=payload)
+    response.raise_for_status()
+
+    return response.json()
+
+
+# Example usage
+if __name__ == "__main__":
+    # Test image URL
+    image_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/4/47/PNG_transparency_demonstration_1.png/280px-PNG_transparency_demonstration_1.png"
+
+    image_url = "https://cloud.toto.com.uy:9000/Fotos/73-2ZY9D63_PINK_GT22663_1.jpg"
+
+    try:
+        start_time = time.time()
+        result = get_image_embedding(image_url)
+        end_time = time.time()
+        elapsed_time = end_time - start_time
+        print(f"Response time: {elapsed_time:.2f} seconds")
+        print(f"Embedding dimension: {result['embedding_dimension']}")
+        print(f"First 5 values: {result['embedding'][:5]}")
+    except requests.exceptions.RequestException as e:
+        print(f"Error: {e}")