File size: 1,995 Bytes
8ba5658 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import requests
from pathlib import Path
from PIL import Image
import io
def get_stored_token():
"""Get the stored HuggingFace token"""
token_path = Path.home() / '.cache/huggingface/token'
if token_path.exists():
with open(token_path, 'r') as f:
return f.read().strip()
return None
# Update API URL to use the inference API endpoint
API_URL = "https://c3g262qlc7cizj5n.us-east4.gcp.endpoints.huggingface.cloud"
token = get_stored_token()
def query(image_path):
# Read image bytes directly
with open(image_path, "rb") as f:
image_bytes = f.read()
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "image/jpeg"
}
# Print some debug info
print(f"Sending file: {image_path}")
print(f"Content-Type: {headers['Content-Type']}")
print(f"Image size: {len(image_bytes)} bytes")
response = requests.post(
API_URL,
headers=headers,
data=image_bytes, # Send raw bytes
verify=True
)
# Add error handling
if response.status_code != 200:
print(f"Response headers: {response.headers}")
print(f"Request headers sent: {response.request.headers}")
return f"Error: {response.status_code}, {response.text}"
try:
return response.json()
except requests.exceptions.JSONDecodeError:
return f"Error decoding JSON. Raw response: {response.text}"
# Test with an image
if __name__ == "__main__":
# Option 1: Test with specific image
image_path = Path("images/20250121_gauge_0001.jpg")
# Option 2: Test with first image found in directory
# TRAIN_IMAGES_DIR = Path("images")
# image_path = next(TRAIN_IMAGES_DIR.glob('*.jpg'))
if not image_path.exists():
print(f"Error: Image not found at {image_path}")
exit(1)
print(f"Testing with image: {image_path}")
result = query(image_path)
print("\nAPI Response:")
print(result) |