Álisson Mello da Rosa commited on
Commit
f0a6620
·
1 Parent(s): 6a32fb9

Correção: removendo exemplos inexistentes para evitar erro de path

Browse files
Files changed (1) hide show
  1. app.py +13 -21
app.py CHANGED
@@ -1,30 +1,22 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
- from PIL import Image
4
 
5
- # 1. Carregamos o modelo globalmente para melhor performance
6
  classificador = pipeline("image-classification", model="apple/mobilevit-small")
7
 
8
- def classificar_imagem(img):
9
- if img is None:
10
- return None
11
-
12
- # 2. Rodar o modelo na imagem vinda da interface
13
- predicoes = classificador(img)
14
-
15
- # 3. Formatamos o resultado para o componente Label do Gradio
16
- # O Gradio espera um dicionário no formato {"Label": confianca}
17
- return {p['label']: p['score'] for p in predicoes}
18
 
19
- # 4. Construção da Interface
20
- app = gr.Interface(
21
- fn=classificar_imagem,
22
- inputs=gr.Image(type="pil", label="Carregue uma imagem"),
23
- outputs=gr.Label(num_top_classes=5, label="Resultados"),
24
- title="Classificador MobileViT (Apple)",
25
- description="Envie uma imagem para que o modelo MobileViT identifique o conteúdo. Modelo leve e otimizado para dispositivos móveis.",
26
- examples=["exemplo_cachorro.jpg"] # Opcional: você pode subir exemplos no Space
27
  )
28
 
29
  if __name__ == "__main__":
30
- app.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
 
3
 
4
+ # Carrega o modelo da Apple
5
  classificador = pipeline("image-classification", model="apple/mobilevit-small")
6
 
7
+ def predict(img):
8
+ predictions = classificador(img)
9
+ # Retorna o dicionário de labels e scores para o Gradio criar as barras
10
+ return {p['label']: p['score'] for p in predictions}
 
 
 
 
 
 
11
 
12
+ # Interface limpa sem a lista de exemplos que está causando erro
13
+ demo = gr.Interface(
14
+ fn=predict,
15
+ inputs=gr.Image(type="pil"),
16
+ outputs=gr.Label(num_top_classes=5),
17
+ title="Classificador MobileViT - Apple",
18
+ description="Envie uma imagem para identificar objetos usando o modelo MobileViT."
 
19
  )
20
 
21
  if __name__ == "__main__":
22
+ demo.launch()