sensei-ml commited on
Commit
107d545
·
verified ·
1 Parent(s): 589a962

Update model/model.py

Browse files
Files changed (1) hide show
  1. model/model.py +1 -1
model/model.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from PIL import Image
6
 
7
  def predict(image):
8
- # Cargar el modelo
9
  model = AutoModelForImageClassification.from_pretrained('sensei-ml/concrete_crack_images_classification')
10
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
11
  model.eval()
 
5
  from PIL import Image
6
 
7
  def predict(image):
8
+ # Load the model
9
  model = AutoModelForImageClassification.from_pretrained('sensei-ml/concrete_crack_images_classification')
10
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
11
  model.eval()