Update app.py

app.py

@@ -10,8 +10,8 @@ from model import generate2,ClipCaptionModel
 from engine import inference
 
 
-# model_trained = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
-# model_trained.load_state_dict(torch.load('model_trained.pth',map_location=torch.device('cpu')))
+model_trained = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+model_trained.load_state_dict(torch.load('model_trained.pth',map_location=torch.device('cpu')))
 image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 
@@ -44,12 +44,12 @@ model = model.eval()
 
 coco_model = ClipCaptionModel(prefix_length)
 coco_model.load_state_dict(torch.load('COCO_model.h5',map_location=torch.device('cpu')))
-model = model.eval() 
+# model = model.eval() 
 
 
 def ui():
  st.markdown("# Image Captioning")
- # st.markdown("## Done By- Vageesh, Rushil and Girish")
+ # st.markdown("## Done By- Vageesh and Rushil")
  uploaded_file = st.file_uploader("Upload an Image", type=['png', 'jpeg', 'jpg'])
 
  if uploaded_file is not None: