Update app.py

app.py

@@ -599,7 +599,8 @@ if task == "Image Forgery Detection":
     
     if uploaded_file:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Uploaded Image", use_container_width=True)
+        # Fixed size display - adjust width as needed (300-600 pixels work well)
+        st.image(image, caption="Uploaded Image", width=400)
         
         # Original prediction
         prepared_image = prepare_image_for_forgery(image).reshape(-1, 128, 128, 3)
@@ -659,7 +660,8 @@ if task == "Image Forgery Detection":
                         analysis_buffer = create_noise_analysis(image)
                         filename = "noise_analysis.png"
                     
-                    st.image(analysis_buffer, caption=f"{analysis_type} Results", use_container_width=True)
+                    # Fixed size for analysis results
+                    st.image(analysis_buffer, caption=f"{analysis_type} Results", width=500)
                     
                     # Detailed results based on analysis type
                     if analysis_type == "Copy-Move Detection":
@@ -716,7 +718,8 @@ elif task == "Deepfake Image Detection":
         with open("temp_image.jpg", "wb") as f:
             f.write(uploaded_file.getbuffer())
 
-        st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
+        # Fixed size display for deepfake detection
+        st.image(uploaded_file, caption="Uploaded Image", width=400)
         model = load_deepfake_image_model()
         result = predict_deepfake_image("temp_image.jpg", model)
 
@@ -726,7 +729,7 @@ elif task == "Deepfake Image Detection":
             st.error("Prediction: Fake")
 
         os.remove("temp_image.jpg")
-
+        
 if task == "Video Forgery Detection":
     uploaded_file = st.file_uploader("Upload a video", type=['mp4', 'avi', 'mov', 'mkv'])