Spaces:
Build error
Build error
Commit ·
dfc2140
1
Parent(s): ad2a517
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ def predict(image):
|
|
| 8 |
classifier = pipeline(task="image-classification")
|
| 9 |
preds = classifier(image)
|
| 10 |
preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
|
| 11 |
-
return preds
|
| 12 |
|
| 13 |
description = """
|
| 14 |
"""
|
|
@@ -16,9 +16,9 @@ description = """
|
|
| 16 |
gr.Interface(
|
| 17 |
fn=predict,
|
| 18 |
inputs=[
|
| 19 |
-
gr.
|
| 20 |
],
|
| 21 |
-
outputs=
|
| 22 |
title="Comparateur d'image",
|
| 23 |
description=description
|
| 24 |
).launch()
|
|
|
|
| 8 |
classifier = pipeline(task="image-classification")
|
| 9 |
preds = classifier(image)
|
| 10 |
preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
|
| 11 |
+
return {"predictions": preds}
|
| 12 |
|
| 13 |
description = """
|
| 14 |
"""
|
|
|
|
| 16 |
gr.Interface(
|
| 17 |
fn=predict,
|
| 18 |
inputs=[
|
| 19 |
+
gr.inputs.Image(label="Image to classify", type="pil"),
|
| 20 |
],
|
| 21 |
+
outputs=gr.outputs.JSON(),
|
| 22 |
title="Comparateur d'image",
|
| 23 |
description=description
|
| 24 |
).launch()
|