Update app.py

app.py

@@ -56,13 +56,6 @@ model2.config.id2label = mapping
 model3 = models.resnet101(pretrained=False)
 model3.fc = torch.nn.Linear(2048, len(genres))
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model1.to(device)
-model2.to(device)
-model3.to(device)
-model_gen.to(device)
-device
-
 
 # In[5]:
 
@@ -93,23 +86,14 @@ class Multimodal(torch.nn.Module):
         output = torch.add(output, image_output)
         return output
 
-
-# In[6]:
-
-
-model = Multimodal(model1, model2, model3)
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model.to(device)
-device
-
-
 # **_PLEASE INSTALL THE MODEL CHECKPOINT FROM THE LINK IN README.txt_**
 
 # In[7]:
 
-
+model = Multimodal(model1, model2, model3)
 model.load_state_dict(torch.load('multimodel.pt'))
 model.eval()
+device = torch.device('cpu')
 
 
 # In[8]: