Spaces:
Runtime error
Runtime error
File size: 2,643 Bytes
16d3463 6053557 16d3463 |
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 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import requests
from urllib.request import urlopen
import base64
import os
def fetch_image(url):
"""
Fetch image data from a URL.
"""
return urlopen(url).read()
def encode_image_to_base64(img_data):
"""
Encode image bytes to a base64 string.
"""
return base64.b64encode(img_data).decode("utf-8")
def send_prediction_request(base64_image, server_url):
"""
Send a single base64 image to the prediction API and retrieve predictions.
"""
try:
response = requests.post(f"{server_url}/predict", json={"image": base64_image})
return response
except requests.exceptions.RequestException as e:
print(f"Error connecting to the server: {e}")
return None
def send_batch_prediction_request(base64_images, server_url):
"""
Send a batch of base64 images to the prediction API and retrieve predictions.
"""
try:
response = requests.post(
f"{server_url}/predict", json=[{"image": img} for img in base64_images]
)
return response
except requests.exceptions.RequestException as e:
print(f"Error connecting to the server: {e}")
return None
def main():
# Server URL (default or from environment)
server_url = os.getenv("SERVER_URL", "http://localhost:8080")
# Example URLs for testing
image_urls = [
"https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png"
]
# Fetch and encode images
try:
print("Fetching and encoding images...")
base64_images = [encode_image_to_base64(fetch_image(url)) for url in image_urls]
print("Images fetched and encoded successfully.")
except Exception as e:
print(f"Error fetching or encoding images: {e}")
return
# Test single image prediction
try:
print("\n--- Single Image Prediction ---")
single_response = send_prediction_request(base64_images[0], server_url)
if single_response and single_response.status_code == 200:
predictions = single_response.json().get("predictions", [])
if predictions:
print("Top 5 Predictions:")
for pred in predictions:
print(f"{pred['label']}: {pred['probability']:.2%}")
else:
print("No predictions returned.")
elif single_response:
print(f"Error: {single_response.status_code}")
print(single_response.text)
except Exception as e:
print(f"Error sending single prediction request: {e}")
if __name__ == "__main__":
main()
|