Spaces:
Sleeping
Sleeping
File size: 681 Bytes
dfec65b 2c92669 dfec65b 87fe2e2 dfec65b 87fe2e2 ad2a517 2c92669 ea9eea4 3ab5bd0 0c5d83d dfec65b dfc2140 dfec65b 2c92669 0c5d83d dfec65b 0c5d83d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import torch
import gradio as gr
import json
from transformers import pipeline
device = "cuda" if torch.cuda.is_available() else "cpu"
def predict(image):
classifier = pipeline(task="image-classification")
preds = classifier(image)
preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
prediction_json = json.dumps(preds)
prediction_dict = json.loads(prediction_json)
return prediction_dict
description = """
"""
gr.Interface(
fn=predict,
inputs=[
gr.inputs.Image(label="Image to classify", type="pil"),
],
outputs="label",
title="Comparateur d'image",
description=description
).launch()
|