Update pages/27_Inception_model.py

pages/27_Inception_model.py

@@ -7,6 +7,7 @@ from torch.utils.data import DataLoader
 import streamlit as st
 import matplotlib.pyplot as plt
 import numpy as np
+from PIL import Image
 
 # Define the Inception Module
 class InceptionModule(nn.Module):
@@ -77,7 +78,7 @@ st.title("InceptionNet Image Classifier")
 
 uploaded_file = st.file_uploader("Choose an image...", type="jpg")
 if uploaded_file is not None:
-    image = plt.imread(uploaded_file)
+    image = Image.open(uploaded_file)
     st.image(image, caption='Uploaded Image', use_column_width=True)
     
     image = transform(image).unsqueeze(0).to(device)