Update app.py

app.py

@@ -541,13 +541,13 @@ def nvidia_deepfake_detection_app():
     if uploaded_file is not None:
         image_bytes = uploaded_file.getvalue()
         image = Image.open(BytesIO(image_bytes))
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption="Uploaded Image", use_container_width=True)
         
         col1, col2 = st.columns([2, 1])
         
         with col1:
             # Display original image
-            st.image(image, caption="Uploaded Image", use_column_width=True)
+            st.image(image, caption="Uploaded Image", use_container_width=True)
         
         with col2:
             # Placeholder for detection results