Spaces:

Alan7
/

ProyectoComputerVision

Runtime error

Alan7 commited on May 30, 2024

Commit

057e300

verified ·

1 Parent(s): f203d44

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 # Define el modelo y carga los pesos guardados
 model = efficientnet_b0(weights=EfficientNet_B0_Weights.DEFAULT)
 model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=101)
-model.load_state_dict(torch.load('./Model_Food_ProyectoIA'), map_location="cpu")
 model.eval()  # Poner el modelo en modo evaluación

 # Define el modelo y carga los pesos guardados
 model = efficientnet_b0(weights=EfficientNet_B0_Weights.DEFAULT)
 model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=101)
+model.load_state_dict(torch.load('./Model_Food_ProyectoIA'), map_location=torch.device('cpu'))
 model.eval()  # Poner el modelo en modo evaluación