Hugging Face's logo

Spaces:
AlejandrOSM1
/
Truck_classifier
Runtime error

App Files Community
AlejandrOSM1 commited on
Commit
36220ad
·
verified ·
1 Parent(s): c5481bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -5,8 +5,7 @@ from fastai.vision.all import *
5
  from huggingface_hub import login, from_pretrained_fastai
6
 
7
  # Cargar el modelo
8
- learn = pipeline("image-classification", model="AlejandrOSM1/Clasificador_de_camiones_Definitivo_Resnet18")
9
-
10
  # Obtener etiquetas del modelo
11
  labels = learn.dls.vocab
12
 
 
5
  from huggingface_hub import login, from_pretrained_fastai
6
 
7
  # Cargar el modelo
8
+ learn = from_pretrained_fastai("AlejandrOSM1/")
 
9
  # Obtener etiquetas del modelo
10
  labels = learn.dls.vocab
11