Update app.py

app.py

@@ -44,6 +44,15 @@ def process_image():
         # Build the prompt
         prompt = f"Remove the {object_type} from the image and fill the area naturally."
 
+        # Disable all safety settings
+        safety_settings = {
+            "HARM_CATEGORY_HARASSMENT": "BLOCK_NONE",
+            "HARM_CATEGORY_HATE_SPEECH": "BLOCK_NONE",
+            "HARM_CATEGORY_SEXUALLY_EXPLICIT": "BLOCK_NONE",
+            "HARM_CATEGORY_DANGEROUS_CONTENT": "BLOCK_NONE",
+            "HARM_CATEGORY_CIVIC_INTEGRITY": "BLOCK_NONE"
+        }
+
         # Generate content
         response = model.generate_content(
             [
@@ -56,9 +65,7 @@ def process_image():
                 "top_k": 40,
                 "max_output_tokens": 8192,
             },
-            safety_settings={
-                "HARM_CATEGORY_CIVIC_INTEGRITY": "BLOCK_NONE"
-            }
+            safety_settings=safety_settings
         )
 
         # Process response