Update app.py

app.py

@@ -18,7 +18,7 @@ def create_caption_transformer(img):
     return caption_text
 
 
-model = VisionEncoderDecoderModel.from_pretrained(os.getcwd() + '/transformer').to('cpu')
+model = VisionEncoderDecoderModel.from_pretrained(os.getcwd()).to('cpu')
 feature_extractor = AutoFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
 tokenizer = AutoTokenizer.from_pretrained('gpt2')
 iface = gr.Interface(fn=create_caption_transformer,