Update app.py

app.py

@@ -18,17 +18,18 @@ if uploaded_image is not None:
 
     # Convert the uploaded file to a PIL Image object
     image = Image.open(uploaded_image).convert('RGB')
-    image = image.resize((178,218))
+    image = image.resize((178, 218))
 
-    # Convert the PIL Image object to a NumPy array
-    image = np.array(image)/255.0
+    # Display the uploaded image
+    st.image(image, caption="Imagen cargada", use_column_width=True)
 
-    image = np.expand_dims(image,axis=0)
+    # Convert the PIL Image object to a NumPy array
+    image_array = np.array(image) / 255.0
+    image_array = np.expand_dims(image_array, axis=0)
 
     # Make a prediction
-    prediction = model.predict(image)
+    prediction = model.predict(image_array)
 
     # Display the prediction
     st.write("La predicción es:")
-    st.write(prediction)
-    
+    st.write(prediction)