Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -12,9 +12,11 @@ def classify_image_with_text(text, image):
|
|
12 |
inputs = processor(
|
13 |
text=["a photo of a man", "a photo of a woman"], images=image, return_tensors="pt", padding=True
|
14 |
)
|
15 |
-
|
|
|
16 |
scores = outputs.logits_per_image
|
17 |
-
|
|
|
18 |
return labels, scores
|
19 |
|
20 |
# Créer l'interface Gradio avec l'API de Gradio Blocks
|
|
|
12 |
inputs = processor(
|
13 |
text=["a photo of a man", "a photo of a woman"], images=image, return_tensors="pt", padding=True
|
14 |
)
|
15 |
+
outputs = model(**inputs)
|
16 |
+
# Récupérer les scores et les libellés
|
17 |
scores = outputs.logits_per_image
|
18 |
+
predictions = outputs.predictions
|
19 |
+
labels = [processor.tokenizer.decode([p]) for p in predictions]
|
20 |
return labels, scores
|
21 |
|
22 |
# Créer l'interface Gradio avec l'API de Gradio Blocks
|