Update app.py

app.py

@@ -1,3 +1,4 @@
+import numpy as np
 from PIL import Image
 import streamlit as st
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
@@ -44,10 +45,10 @@ def main():
     # check if file has been uploaded
     if uploaded_file is not None:
         # load the image
-        image = Image.open(uploaded_file).convert("RGB")
-
+        image = np.array(Image.open(uploaded_file).resize((224, 224)))
+        
         # context as prompt
-        prompt = generate_captions(uploaded_file)
+        prompt = generate_captions(image)
         st.write("The Context is:", prompt)
 
         # display the image