Update app.py

app.py

@@ -36,7 +36,7 @@ class Attention(nn.Module):
 # -----------------------
 # Pre-trained VQA Model
 # -----------------------
-class VQAModel(nn.Module):
+class PretrainedVQAModel(nn.Module):
     def __init__(self, vocab_size, embedding_dim=256, lstm_units=256, attention_dim=256, max_seq_len=30):
         super(PretrainedVQAModel, self).__init__()
         self.vocab_size = vocab_size
@@ -133,7 +133,7 @@ def load_model(model_path, word_to_idx_path, idx_to_word_path, device='cpu'):
         idx_to_word = torch.load(idx_to_word_path, map_location=device)
         
         # Khởi tạo mô hình
-        model = VQAModel(vocab_size=len(word_to_idx))
+        model = PretrainedVQAModel(vocab_size=len(word_to_idx))
         model.load_state_dict(torch.load(model_path, map_location=device))
         model.to(device)
         model.eval()