fixed error

src/streamlit_app.py

@@ -21,6 +21,7 @@ import torchvision.transforms as transforms
 from PIL import Image
 from resnet_model import MonkeyResNet
 from data_loader import get_data_loaders
+import io
 
 
 # Ensure the parent directory is in the system path for module imports
@@ -100,6 +101,7 @@ with tab2:
         raw_bytes = uploaded_file.read()
         image = Image.open(io.BytesIO(raw_bytes)).convert("RGB")
         st.image(image, caption="Uploaded Image", width=300)
+        input_tensor = preprocess_image(image)
 
         input_tensor = preprocess_image(image)