| import requests | |
| import os | |
| import gradio as gr | |
| API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-50-panoptic" | |
| headers = {"Authorization": "Bearer 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"} | |
| def image_classifier(inp): | |
| return {'cat': 0.3, 'dog': 0.7} | |
| demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label") | |
| demo.launch() | |
| # def query(filename): | |
| # with open(filename, "rb") as f: | |
| # data = f.read() | |
| # response = requests.post(API_URL, headers=headers, data=data) | |
| # return response.json() | |
| # # train = os.listdir("./") | |
| # # print(train) | |
| # output = query("./09_truck.jpg") | |
| # print(output) |