MerchFlow-AI / check_basic.py
Gaurav vashistha
Final Release: Updated README and Sync
842ca78
raw
history blame contribute delete
616 Bytes
import os
from huggingface_hub import InferenceClient
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("HF_TOKEN")
client = InferenceClient(api_key=api_key)
print(f"Testing token with microsoft/resnet-50")
try:
# Pass the URL directly as the input (InferenceClient handles URLs for image tasks)
result = client.image_classification(
model="microsoft/resnet-50",
image="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
)
print("Success:", result)
except Exception as e:
print("Failed:", e)