Spaces:
Running
Running
Update feifeilib/feifeichat.py
Browse files- 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 |
-
|
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 |
-
|
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)
|