Update app.py

app.py

@@ -77,10 +77,10 @@ from transformers import ViTFeatureExtractor, ViTModel
 vit_feat_extract = ViTFeatureExtractor("google/vit-base-patch16-224-in21k")
 
 def answer_question(image, question):
-    image.save('sample_img.jpg')
+    image.save('sample_img.png')
 
     # Extracting features from the image
-    img, boxes, tokenized_words = create_features(image_path='sample_img.jpg',
+    img, boxes, tokenized_words = create_features(image_path='sample_img.png',
                                                   tokenizer=tokenizer,
                                                   target_size=target_size,
                                                   max_seq_length=max_seq_len,