Update app.py

app.py

@@ -6,7 +6,7 @@ import torchvision.transforms as transforms
 from transformers import VisionEncoderDecoderModel, AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Load the models
-caption_model = VisionEncoderDecoderModel.from_pretrained('/content/drive/MyDrive/ICModel')
+caption_model = VisionEncoderDecoderModel.from_pretrained('Mayada/AIC-transformer')
 caption_tokenizer = AutoTokenizer.from_pretrained('aubmindlab/bert-base-arabertv02')
 question_model = AutoModelForSeq2SeqLM.from_pretrained("Mihakram/AraT5-base-question-generation")
 question_tokenizer = AutoTokenizer.from_pretrained("Mihakram/AraT5-base-question-generation")