|
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} |
|
|
|
def query(filename): |
|
with open(filename, "rb") as f: |
|
data = f.read() |
|
response = requests.post(API_URL, headers=headers, data=data) |
|
return response.json() |
|
|
|
def rb(img): |
|
response = requests.post(API_URL, headers=headers, data=bytes(img.getdata())) |
|
return response.json() |
|
|
|
|
|
|
|
output = query("./09_truck.jpg") |
|
|
|
|
|
|
|
inputs = gr.inputs.Image(type="pil", label="Upload an image") |
|
demo = gr.Interface(fn=rb, inputs=inputs, outputs="json") |
|
demo.launch() |