Spaces:

dduy193
/

Movie-Genres-Multilabel_MultiPoro

Runtime error

dduy193 commited on Dec 25, 2023

Commit

02dd597

1 Parent(s): f32a873

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ class Multimodal(torch.nn.Module):
 # In[7]:
 model = Multimodal(model1, model2, model3)
-model = model.load_state_dict(torch.load('multimodel.pt'), map_location=torch.device('cpu'))
 model.to(device)
 model.eval()
 model_gen.to(device)

 # In[7]:
 model = Multimodal(model1, model2, model3)
+model.load_state_dict(torch.load('multimodel.pt', map_location=torch.device('cpu')))
 model.to(device)
 model.eval()
 model_gen.to(device)