Alan7 commited on
Commit
a707b53
verified
1 Parent(s): 057e300

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,7 +8,10 @@ import gradio as gr
8
  # Define el modelo y carga los pesos guardados
9
  model = efficientnet_b0(weights=EfficientNet_B0_Weights.DEFAULT)
10
  model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=101)
11
- model.load_state_dict(torch.load('./Model_Food_ProyectoIA'), map_location=torch.device('cpu'))
 
 
 
12
 
13
  model.eval() # Poner el modelo en modo evaluaci贸n
14
 
 
8
  # Define el modelo y carga los pesos guardados
9
  model = efficientnet_b0(weights=EfficientNet_B0_Weights.DEFAULT)
10
  model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=101)
11
+
12
+ #model.load_state_dict(torch.load('./Model_Food_ProyectoIA'), map_location=torch.device('cpu'))
13
+ model.load_state_dict(torch.load('./Model_Food_ProyectoIA', map_location=torch.device('cpu')))
14
+
15
 
16
  model.eval() # Poner el modelo en modo evaluaci贸n
17