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: