Update app.py

app.py

@@ -12,8 +12,8 @@ labels = [
     "No es un utensilio de laboratorio"
 ]
 
-# Cargar el modelo directamente (debe estar en la misma carpeta o en models/)
-learn = vision_learner(dls=None, arch=resnet34)  # dls=None porque no hay dataset
+# Cargar el modelo directamente, indicando n_out porque no hay dataset
+learn = vision_learner(dls=None, arch=resnet34, n_out=len(labels))
 learn.load("model_lab")  # si lo pusiste en models/, usa "models/model_lab"
 
 def predict(img):
@@ -28,4 +28,4 @@ demo = gr.Interface(
     title="Lab Utensils Classifier"
 )
 
-demo.launch()
+demo.launch()