mestrevh commited on
Commit
3d842de
·
1 Parent(s): dadd386

Add the files

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +1 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ computer-vision-beans/
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Carregar o modelo diretamente do Hugging Face Hub
5
- classifier = pipeline("image-classification", model="mestrevh/computer-vision-cifar-10")
6
 
7
  # Função de classificação
8
  def predict_image(image):
 
2
  from transformers import pipeline
3
 
4
  # Carregar o modelo diretamente do Hugging Face Hub
5
+ classifier = pipeline("image-classification", model="mestrevh/computer-vision-beans")
6
 
7
  # Função de classificação
8
  def predict_image(image):