Update app.py

app.py

@@ -27,7 +27,7 @@ file_paths = []
 # Function to get Gradio NSFW prediction
 def get_gradio_nsfw_prediction(image_path):
     result = client_nsfw.predict(image_path, api_name="/predict")
-    return result["label"]  # Assuming the label is the prediction result
+    return result  # Return the entire dictionary
 
 # Define the function to generate a download link for a file
 def get_binary_file_downloader_html(file_path, label="Download"):
@@ -94,7 +94,7 @@ def display_images(images):
             st.write(f"**Good value:** {good_values[i]}")
             st.write(f"**Timestamp:** {timestamp}")
 
-            if st.session_state.blur_option and gradio_nsfw_prediction.lower() == "nsfw":
+            if st.session_state.blur_option and gradio_nsfw_prediction.get("label", "").lower() == "nsfw":
                 blurred_img = apply_blur(file_path)
                 st.image(blurred_img, caption=os.path.basename(file_path), use_column_width=True)
             else: