Update app.py

app.py

@@ -121,7 +121,10 @@ with right_column:
         model_s.load_state_dict(torch.load('celeb-model-unlearned.pth', map_location=torch.device('cpu')))
         model_s.eval()
 
-        uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+        uploaded_file2 = st.file_uploader("Choose image...", type=["jpg", "jpeg", "png"])
+        image2 = Image.open(uploaded_file2)
+        st.image(image2, caption='Uploaded Image.', width=300)        
+
                 # Perform inference
         st.write("Performing inference...")