Spaces:
Runtime error
Runtime error
Commit
•
a43bb6f
1
Parent(s):
f31c256
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ device = 'cpu'
|
|
11 |
tokenizer = load("tokenizer_v3.joblib")
|
12 |
feature_extractor = load("feature_extractor_v3.joblib")
|
13 |
|
14 |
-
model = VisionEncoderDecoderModel.from_pretrained("dumperize/movie-picture-captioning")
|
15 |
-
|
16 |
model.load_state_dict(torch.load("model_weights_i2t_fin.pt", map_location=torch.device('cpu')))
|
17 |
# model.eval()
|
18 |
|
|
|
11 |
tokenizer = load("tokenizer_v3.joblib")
|
12 |
feature_extractor = load("feature_extractor_v3.joblib")
|
13 |
|
14 |
+
# model = VisionEncoderDecoderModel.from_pretrained("dumperize/movie-picture-captioning")
|
15 |
+
model = load("img2txt_v4.joblib")
|
16 |
model.load_state_dict(torch.load("model_weights_i2t_fin.pt", map_location=torch.device('cpu')))
|
17 |
# model.eval()
|
18 |
|