Update feifeilib/feifeichat.py

feifeilib/feifeichat.py

@@ -10,6 +10,7 @@ client = InferenceClient(api_key=os.getenv('HF_TOKEN'))
 client.headers["x-use-cache"] = "0"
 api_key = os.getenv("MISTRAL_API_KEY")
 Mistralclient = Mistral(api_key=api_key)
+from numba import jit
 
 def encode_image(image_path):
     """Encode the image to base64."""
@@ -30,6 +31,7 @@ def encode_image(image_path):
         print(f"Error: {e}")
         return None
 
+@jit
 def feifeiprompt(feifei_select=True, message_text="", history=""):
     input_prompt = []
     if message_text.startswith("画") or message_text.startswith("draw"):
@@ -58,6 +60,7 @@ def feifeiprompt(feifei_select=True, message_text="", history=""):
 
     return input_prompt
 
+@jit
 def feifeiimgprompt(message_files, message_text, image_mod):
     message_file = message_files[0]
     base64_image = encode_image(message_file)