Kaynağa Gözat

Minor changes

oobabooga 2 yıl önce
ebeveyn
işleme
104212529f

+ 3 - 6
extensions/elevenlabs_tts/script.py

@@ -1,13 +1,11 @@
+import re
 from pathlib import Path
 from pathlib import Path
 
 
 import gradio as gr
 import gradio as gr
+import modules.shared as shared
 from elevenlabslib import ElevenLabsUser
 from elevenlabslib import ElevenLabsUser
 from elevenlabslib.helpers import save_bytes_to_path
 from elevenlabslib.helpers import save_bytes_to_path
 
 
-import re
-
-import modules.shared as shared
-
 params = {
 params = {
     'activate': True,
     'activate': True,
     'api_key': '12345',
     'api_key': '12345',
@@ -54,7 +52,6 @@ def refresh_voices():
         return
         return
 
 
 def remove_surrounded_chars(string):
 def remove_surrounded_chars(string):
-    # regexp is way faster than repeated string concatenation!
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     return re.sub('\*[^\*]*?(\*|$)','',string)
     return re.sub('\*[^\*]*?(\*|$)','',string)
@@ -113,4 +110,4 @@ def ui():
     voice.change(lambda x: params.update({'selected_voice': x}), voice, None)
     voice.change(lambda x: params.update({'selected_voice': x}), voice, None)
     api_key.change(lambda x: params.update({'api_key': x}), api_key, None)
     api_key.change(lambda x: params.update({'api_key': x}), api_key, None)
     connect.click(check_valid_api, [], connection_status)
     connect.click(check_valid_api, [], connection_status)
-    connect.click(refresh_voices, [], voice)
+    connect.click(refresh_voices, [], voice)

+ 3 - 5
extensions/sd_api_pictures/script.py

@@ -4,13 +4,12 @@ import re
 from pathlib import Path
 from pathlib import Path
 
 
 import gradio as gr
 import gradio as gr
+import modules.chat as chat
+import modules.shared as shared
 import requests
 import requests
 import torch
 import torch
 from PIL import Image
 from PIL import Image
 
 
-import modules.chat as chat
-import modules.shared as shared
-
 torch._C._jit_set_profiling_mode(False)
 torch._C._jit_set_profiling_mode(False)
 
 
 # parameters which can be customized in settings.json of webui  
 # parameters which can be customized in settings.json of webui  
@@ -32,7 +31,6 @@ picture_response = False # specifies if the next model response should appear as
 pic_id = 0
 pic_id = 0
 
 
 def remove_surrounded_chars(string):
 def remove_surrounded_chars(string):
-    # regexp is way faster than repeated string concatenation!
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     return re.sub('\*[^\*]*?(\*|$)','',string)
     return re.sub('\*[^\*]*?(\*|$)','',string)
@@ -186,4 +184,4 @@ def ui():
 
 
     force_btn.click(force_pic)
     force_btn.click(force_pic)
     generate_now_btn.click(force_pic)
     generate_now_btn.click(force_pic)
-    generate_now_btn.click(eval('chat.cai_chatbot_wrapper'), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)
+    generate_now_btn.click(eval('chat.cai_chatbot_wrapper'), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)

+ 4 - 5
extensions/send_pictures/script.py

@@ -2,12 +2,11 @@ import base64
 from io import BytesIO
 from io import BytesIO
 
 
 import gradio as gr
 import gradio as gr
-import torch
-from transformers import BlipForConditionalGeneration, BlipProcessor
-from PIL import Image
-
 import modules.chat as chat
 import modules.chat as chat
 import modules.shared as shared
 import modules.shared as shared
+import torch
+from PIL import Image
+from transformers import BlipForConditionalGeneration, BlipProcessor
 
 
 # If 'state' is True, will hijack the next chat generation with
 # If 'state' is True, will hijack the next chat generation with
 # custom input text given by 'value' in the format [text, visible_text]
 # custom input text given by 'value' in the format [text, visible_text]
@@ -54,4 +53,4 @@ def ui():
     picture_select.upload(eval(function_call), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)
     picture_select.upload(eval(function_call), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)
 
 
     # Clear the picture from the upload field
     # Clear the picture from the upload field
-    picture_select.upload(lambda : None, [], [picture_select], show_progress=False)
+    picture_select.upload(lambda : None, [], [picture_select], show_progress=False)

+ 3 - 5
extensions/silero_tts/script.py

@@ -1,12 +1,11 @@
+import re
 import time
 import time
 from pathlib import Path
 from pathlib import Path
 
 
 import gradio as gr
 import gradio as gr
-import torch
-import re
-
 import modules.chat as chat
 import modules.chat as chat
 import modules.shared as shared
 import modules.shared as shared
+import torch
 
 
 torch._C._jit_set_profiling_mode(False)
 torch._C._jit_set_profiling_mode(False)
 
 
@@ -47,7 +46,6 @@ def load_model():
 model = load_model()
 model = load_model()
 
 
 def remove_surrounded_chars(string):
 def remove_surrounded_chars(string):
-    # regexp is way faster than repeated string concatenation!
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string'
     return re.sub('\*[^\*]*?(\*|$)','',string)
     return re.sub('\*[^\*]*?(\*|$)','',string)
@@ -163,4 +161,4 @@ def ui():
     autoplay.change(lambda x: params.update({"autoplay": x}), autoplay, None)
     autoplay.change(lambda x: params.update({"autoplay": x}), autoplay, None)
     voice.change(lambda x: params.update({"speaker": x}), voice, None)
     voice.change(lambda x: params.update({"speaker": x}), voice, None)
     v_pitch.change(lambda x: params.update({"voice_pitch": x}), v_pitch, None)
     v_pitch.change(lambda x: params.update({"voice_pitch": x}), v_pitch, None)
-    v_speed.change(lambda x: params.update({"voice_speed": x}), v_speed, None)
+    v_speed.change(lambda x: params.update({"voice_speed": x}), v_speed, None)