Saad0KH commited on
Commit
01b6be2
·
verified ·
1 Parent(s): 472e296

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- # Récupérer les scores et les libellés
 
16
  scores = outputs.logits_per_image
17
- labels = processor.tokenizer.decode(outputs.prediction)
 
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