mestrevh commited on
Commit
d1f49e2
·
1 Parent(s): ffd5b1d

modified the app.py

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +4 -4
app.py CHANGED
@@ -11,8 +11,8 @@ def classify_bean(image):
11
  if not isinstance(image, Image.Image):
12
  image = Image.open(image)
13
 
14
- # Ativar padding no pré-processamento
15
- result = classifier(image, padding=True)
16
 
17
  # Retornar o resultado
18
  return result[0]['label'], result[0]['score']
 
11
  if not isinstance(image, Image.Image):
12
  image = Image.open(image)
13
 
14
+ # Realizar a classificação sem o padding, pois não é necessário para classificação de imagem
15
+ result = classifier(image)
16
 
17
  # Retornar o resultado
18
  return result[0]['label'], result[0]['score']
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- transformers==4.29.0
2
- datasets==2.10.0
3
- torch==2.1.0
4
- gradio==3.33.0
5
  safetensors
 
1
+ transformers
2
+ datasets
3
+ torch
4
+ gradio
5
  safetensors