Update app.py

app.py

@@ -21,7 +21,7 @@ from vocab import Vocabulary
 from model import *
 
 # Paths
-ckpt_dir = "./model-epoch-30.pth"  # Path to the trained model
+ckpt_dir = "./best_model.pth"  # Path to the trained model
 ques_vocab_path = "./question_vocabs.txt"  # Path to question vocabulary
 ans_vocab_path = "./annotation_vocabs.txt"  # Path to answer vocabulary
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -49,7 +49,7 @@ def preprocess_question(question, ques_vocab, max_qu_len):
 
 # Load Model and Checkpoint
 def load_model():
-    model = VQAModel(feature_size=FEATURE_SIZE,
+    model = VQAModel_attn(feature_size=FEATURE_SIZE,
                      ques_vocab_size=ques_vocab.vocabulary_size,
                      ans_vocab_size=ans_vocab.vocabulary_size,
                      word_embed=WORD_EMBED,