aifeifei798 commited on
Commit
74631db
·
verified ·
1 Parent(s): cc08b4d

Update feifeilib/feifeichat.py

Browse files
Files changed (1) hide show
  1. feifeilib/feifeichat.py +3 -3
feifeilib/feifeichat.py CHANGED
@@ -10,7 +10,7 @@ client = InferenceClient(api_key=os.getenv('HF_TOKEN'))
10
  client.headers["x-use-cache"] = "0"
11
  api_key = os.getenv("MISTRAL_API_KEY")
12
  Mistralclient = Mistral(api_key=api_key)
13
- from numba import jit
14
 
15
  def encode_image(image_path):
16
  """Encode the image to base64."""
@@ -31,7 +31,7 @@ def encode_image(image_path):
31
  print(f"Error: {e}")
32
  return None
33
 
34
- @jit
35
  def feifeiprompt(feifei_select=True, message_text="", history=""):
36
  input_prompt = []
37
  if message_text.startswith("画") or message_text.startswith("draw"):
@@ -60,7 +60,7 @@ def feifeiprompt(feifei_select=True, message_text="", history=""):
60
 
61
  return input_prompt
62
 
63
- @jit
64
  def feifeiimgprompt(message_files, message_text, image_mod):
65
  message_file = message_files[0]
66
  base64_image = encode_image(message_file)
 
10
  client.headers["x-use-cache"] = "0"
11
  api_key = os.getenv("MISTRAL_API_KEY")
12
  Mistralclient = Mistral(api_key=api_key)
13
+ #from numba import jit
14
 
15
  def encode_image(image_path):
16
  """Encode the image to base64."""
 
31
  print(f"Error: {e}")
32
  return None
33
 
34
+ #@jit
35
  def feifeiprompt(feifei_select=True, message_text="", history=""):
36
  input_prompt = []
37
  if message_text.startswith("画") or message_text.startswith("draw"):
 
60
 
61
  return input_prompt
62
 
63
+ #@jit
64
  def feifeiimgprompt(message_files, message_text, image_mod):
65
  message_file = message_files[0]
66
  base64_image = encode_image(message_file)