malaria_classifier_api / test_live_api.py
anujjj321's picture
Upload folder using huggingface_hub
f65b509 verified
Raw
History Blame Contribute Delete
1.3 kB
import requests
# ⚠️ REPLACE this URL with your actual Hugging Face Space URL!
# Example format: "https://your-username-malaria-classifier-api.hf.space/predict"
API_URL = "https://<YOUR_HUGGINGFACE_USERNAME>-malaria-classifier-api.hf.space/predict"
# Path to a locally saved cell image you want to test
IMAGE_PATH = "processed_cells/parasite/cell_0.png" # You may need to update this name based on existing files
def test_api():
try:
with open(IMAGE_PATH, 'rb') as f:
# The key 'file' must match the parameter name in our Fastapi app: file: UploadFile = File(...)
files = {'file': (IMAGE_PATH, f, 'image/png')}
print(f"Sending request to {API_URL}...")
response = requests.post(API_URL, files=files)
# Print the result
if response.status_code == 200:
print("✅ Success!")
print("Prediction Data:", response.json())
else:
print(f"❌ Error {response.status_code}: {response.text}")
except FileNotFoundError:
print(f"Please update IMAGE_PATH. Could not find file at: {IMAGE_PATH}")
except Exception as e:
print(f"An error occurred: {e}")
if __name__ == "__main__":
test_api()