Update app.py

app.py

@@ -80,7 +80,7 @@ def answer_question(image, question):
     image.save('sample_img.png')
 
     # Extracting features from the image
-    img, boxes, tokenized_words = create_features(image_path='sample_img.png',
+    img boxes, tokenized_words = create_features(image_path='sample_img.png',
                                                   tokenizer=tokenizer,
                                                   target_size=target_size,
                                                   max_seq_length=max_seq_len,
@@ -105,7 +105,7 @@ def answer_question(image, question):
     ## Tensor tokenized words
     tokenized_words = torch.as_tensor(tokenized_words, dtype=torch.int32)
     
-    img = transforms.ToTensor()(img)
+    img = transforms.ToTensor()(image)
     question = convert_ques_to_token(question = question, tokenizer = tokenizer)
 
     ## Expanding the dimension for inference