few updates

app_dialogue.py

@@ -66,7 +66,7 @@ BAN_TOKENS = (  # For documentation puporse. We are not using this list, it is h
 EOS_STRINGS = ["<end_of_utterance>", "\nUser:"]
 STOP_SUSPECT_LIST = []
 
-GRADIO_LINK = "https://huggingfacem4-ai-meme-generator.hf.space"
+GRADIO_LINK = "https://leyo-ai-meme-generator.hf.space"
 API_TOKEN = os.getenv("HF_AUTH_TOKEN")
 IDEFICS_LOGO = "https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/main/IDEFICS_logo.png"
 
@@ -156,7 +156,7 @@ for image_dir in os.listdir("example_images"):
         tmp_filename = H.preprocess(H.value)
         DEFAULT_IMAGES_TMP_PATH_TO_URL[
             tmp_filename
-        ] = f"https://huggingface.co/spaces/HuggingFaceM4/AI_Meme_Generator/resolve/main/example_images/{image_dir}/{im_path}"
+        ] = f"https://huggingface.co/spaces/Leyo/AI_Meme_Generator/resolve/main/example_images/{image_dir}/{im_path}"
 
 
 # Utils to handle the image markdown display logic
@@ -792,8 +792,12 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         text_at_the_top,
         generation_args,
     ):
-        time.sleep(random.uniform(0, 1))
-        text = client.generate(prompt=query, **generation_args).generated_text
+        # time.sleep(random.uniform(0, 1))
+        try:
+            text = client.generate(prompt=query, **generation_args).generated_text
+        except Exception as e:
+            print(e)
+            text = "BACKEND ERROR"
         if image is not None and text != "":
             meme_image = make_meme_image(
                 image=image,
@@ -1007,4 +1011,4 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         outputs=[template_gallery],
     )
 demo.queue(concurrency_count=8, max_size=40, api_open=False)
-demo.launch(max_threads=400)
+demo.launch(max_threads=400, prevent_thread_lock=True)