| from transformers import AutoFeatureExtractor, AutoModelForImageClassification | |
| extractor = AutoFeatureExtractor.from_pretrained("swww/test") | |
| model = AutoModelForImageClassification.from_pretrained("swww/test") | |
| import requests | |
| API_URL = "https://api-inference.huggingface.co/models/swww/test" | |
| headers = {"Authorization": "Bearer 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"} | |
| def query(filename): | |
| with open(filename, "rb") as f: | |
| data = f.read() | |
| response = requests.post(API_URL, headers=headers, data=data) | |
| return response.json() | |
| output = query("cats.jpg") |