update queuing

app.py

@@ -12,31 +12,31 @@ pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dt
 pipe.to("cuda")
 pipe.enable_attention_slicing()
 
-seed = random.randint(0, 1000000)
+#seed = random.randint(0, 1000000)
 counter = 0
-print(f"SEED IS : {seed}")
+#print(f"SEED IS : {seed}")
 
-def chat(image_in, message, history): #, progress=gr.Progress(track_tqdm=True)):
-    #progress(0, desc="Starting...")
+def chat(image_in, message, history, progress=gr.Progress(track_tqdm=True)):
+    progress(0, desc="Starting...")
     global counter 
-    global seed 
-    img_nm = f"./edited_image_{seed}.png"
+    #global seed 
+    #img_nm = f"./edited_image_{seed}.png"
     counter += 1
-    print(f"seed is : {seed}")
-    print(f"image_in name is :{img_nm}")
+    #print(f"seed is : {seed}")
+    #print(f"image_in name is :{img_nm}")
     
     #if message == "revert": --to add revert functionality later
     if counter > 1:
       # Open the image
-      image_in = Image.open(img_nm) #("./edited_image.png")
+      image_in = Image.open("edited_image.png") #(img_nm) 
     prompt = message #eg - "turn him into cyborg"
     edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
-    edited_image.save(img_nm) #("edited_image.png") #("./edited_image.png")
+    edited_image.save("edited_image.png") # (img_nm) #("./edited_image.png")
     history = history or []
     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? "]
     #Resizing the image for better display
-    response = random.choice(add_text_list) + '<img src="/file=' + img_nm[2:] + '" style="width: 200px; height: 200px;">'
-    #response =  random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 200px; height: 200px;">'
+    #response = random.choice(add_text_list) + '<img src="/file=' + img_nm[2:] + '" style="width: 200px; height: 200px;">'
+    response =  random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 200px; height: 200px;">'
     history.append((message, response))
     return history, history
 
@@ -50,5 +50,5 @@ with gr.Blocks() as demo:
       chatbot = gr.Chatbot() 
     b1.click(chat,[image_in, text_in, state_in], [chatbot, state_in])
 
-#demo.queue()
+demo.queue(concurrency_count=10)
 demo.launch(debug=True, width="80%", height=1500)