Hugging Face's logo

Spaces:
DHEIVER
/
classificador_de_imagem_colonoscopia
Sleeping

App Files Community
DHEIVER commited on
Commit
8dc799b
·
1 Parent(s): 0666586

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import torch
3
  from transformers import ViTFeatureExtractor, ViTForImageClassification
4
  from PIL import Image
5
  import io
 
6
 
7
  # Carregue o modelo ViT
8
  model_name = "mrm8488/vit-base-patch16-224_finetuned-kvasirv2-colonoscopy"
 
3
  from transformers import ViTFeatureExtractor, ViTForImageClassification
4
  from PIL import Image
5
  import io
6
+ import numpy as np
7
 
8
  # Carregue o modelo ViT
9
  model_name = "mrm8488/vit-base-patch16-224_finetuned-kvasirv2-colonoscopy"