Spaces:
Sleeping
Sleeping
Commit
·
76b3196
1
Parent(s):
0545d82
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ train_loader,dataset=get_loader(root_folder='FlickrDataset/Images',annotation_fi
|
|
| 16 |
filepath="ImageCaptioningusingLSTM.pth"
|
| 17 |
from model import CNNtoRNN
|
| 18 |
model=CNNtoRNN(embed_size=256,hidden_size=256,vocab_size=len(dataset.vocab),num_layers=1)
|
| 19 |
-
model.load_state_dict(torch.load(filepath))
|
| 20 |
model.eval()
|
| 21 |
|
| 22 |
def launch(input):
|
|
|
|
| 16 |
filepath="ImageCaptioningusingLSTM.pth"
|
| 17 |
from model import CNNtoRNN
|
| 18 |
model=CNNtoRNN(embed_size=256,hidden_size=256,vocab_size=len(dataset.vocab),num_layers=1)
|
| 19 |
+
model.load_state_dict(torch.load(filepath,map_location='cpu'))
|
| 20 |
model.eval()
|
| 21 |
|
| 22 |
def launch(input):
|