update desc and safety checker

app.py

@@ -10,14 +10,14 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
 
 #Loading from Diffusers Library
 model_id = "timbrooks/instruct-pix2pix"
-pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", safety_checker=None)
+pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16") #, safety_checker=None)
 pipe.to("cuda")
 pipe.enable_attention_slicing()
 
 counter = 0
 
 
-help_text = """ Note: I will try to add the functionality to revert your changes to previous/original image in future versions of space. For now only forward editing is available.
+help_text = """ Note: Functionality to revert your changes to previous/original image can be released in future versions. For now only forward editing is available.
 
 Some notes from the official [instruct-pix2pix](https://huggingface.co/spaces/timbrooks/instruct-pix2pix) Space by the authors  
 and from the official [Diffusers docs](https://huggingface.co/docs/diffusers/main/en/api/pipelines/stable_diffusion/pix2pix) -
@@ -42,53 +42,38 @@ be to the input. This pipeline requires a value of at least `1`. It's possible y
 def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, counter_out, prompt, history, progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Starting...")
     #if message == "revert": --to add revert functionality later
-    print(f"counter:{counter_out}, prompt:{prompt}, img_name:{img_name}")
     if counter_out > 0:
-      # Open the image
-      #image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
       edited_image = pipe(prompt, image=image_hid, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
       if os.path.exists(img_name):
-        print("***Image exists and will be deleted***")
         os.remove(img_name)
-      temp_img_name = img_name[:-4]+str(int(time.time()))+'.png' #img_name[] + str(int(time.time()))
-      print(f"temp_img_name is :{temp_img_name}")
+      temp_img_name = img_name[:-4]+str(int(time.time()))+'.png' 
       # Create a file-like object
       with open(temp_img_name, "wb") as fp:
         # Save the image to the file-like object
         edited_image.save(fp)
       #Get the name of the saved image
       saved_image_name = fp.name
-      print(f"SAVED IMAGE NAME : {saved_image_name}, TMP IMAGE NAME : {temp_img_name}")    
       #edited_image.save(temp_img_name) #, overwrite=True)
       counter_out += 1
     else:
-      print("FIRST PASS")
       seed = random.randint(0, 1000000)
       img_name = f"./edited_image_{seed}.png"
       edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
       if os.path.exists(img_name):
-        print("***First PASS:Image exists and will be deleted***")
         os.remove(img_name)
       with open(img_name, "wb") as fp:
         # Save the image to the file-like object
         edited_image.save(fp)
       #Get the name of the saved image
       saved_image_name2 = fp.name
-      print(f"SAVED IMAGE NAME : {saved_image_name2}, IMAGE NAME : {img_name}")
-      #edited_image.save(img_name) #, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
-      #counter_out += 1
     history = history or []
     #Resizing (or not) the image for better display and adding supportive sample text
     add_text_list = ["There you go", "Enjoy your image!", "Nice work! Wonder what you gonna do next!", "Way to go!", "Does this work for you?", "Something like this?"]
     if counter_out > 0:
-        #response = edited_image
-        print(f"You are in {counter_out} pass")
-        response = random.choice(add_text_list) + '<img src="/file=' + saved_image_name + '">'  #temp_img_name
+        response = random.choice(add_text_list) + '<img src="/file=' + saved_image_name + '">'  
         history.append((prompt, response))
         return history, history, edited_image, temp_img_name, counter_out
     else:
-        #response = edited_image
-        print(f"You are in {counter_out} pass or FIRST Pass")
         response = random.choice(add_text_list) + '<img src="/file=' + saved_image_name2 + '">'   #IMG_NAME
         history.append((prompt, response))
         counter_out += 1
@@ -100,6 +85,10 @@ with gr.Blocks() as demo:
     <p>For faster inference without waiting in the queue, you may duplicate the space and upgrade to GPU in settings.<br/>
     <a href="https://huggingface.co/spaces/ysharma/InstructPix2Pix_Chatbot?duplicate=true">
     <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
+    <b>Note: Please be advised that a safety checker has been implemented in this public space. 
+    Any attempts to generate inappropriate or NSFW images will result in the display of a black screen 
+    as a precautionary measure for the protection of all users. We appreciate your cooperation in 
+    maintaining a safe and appropriate environment for all members of our community.</b>
     <p/>""")
     with gr.Row():
       with gr.Column():