1580158
/
prompt_translator
mirror of https://github.com/ParisNeo/prompt_translator.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/ParisNeo/prompt_translator

pull/31/head
Saifeddine ALOUI 2023-10-28 17:41:54 +02:00
parent fbe0950d5b db40fb7789
commit 72b680f312
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/main.py
View File

@ -353,13 +353,15 @@ class Script(scripts.Script):
self.is_active=active
if not hasattr(self, "translator"):
self.translator = MBartTranslator()
return "ready", self.options.update(visible=True)
if self.is_active:
return "# I am ready", self.options.update(visible=True)
else:
return "After enabling translation, please Wait until I am ready", self.options.update(visible=False)
def set_negative_translate_active(self, negative_translate_active):
"""Sets the is_active attribute and initializes the translator object if not already created.
Also, sets the visibility of the language dropdown to True."""
self.is_negative_translate_active=negative_translate_active
def ui(self, is_img2img):
@ -393,7 +395,7 @@ class Script(scripts.Script):
type="index",
elem_id=self.elem_id("x_type")
)
self.output=gr.Label("After enabling translation, please Wait until I am ready")
self.output=gr.Markdown(value="After enabling translation, please Wait until I am ready", visible=True)
self.enable_translation.change(
self.set_active,
[self.enable_translation],