Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,8 +59,8 @@ llm = Llama.from_pretrained(
|
|
59 |
repo_id="amir22010/fine_tuned_product_marketing_email_gemma_2_9b_q4_k_m", #custom fine tuned model
|
60 |
filename="unsloth.Q4_K_M.gguf", #model file name
|
61 |
cache_dir=os.path.abspath(os.getcwd()),
|
62 |
-
n_ctx=2048,
|
63 |
-
n_batch=126,
|
64 |
verbose=False
|
65 |
)
|
66 |
|
@@ -109,6 +109,7 @@ async def greet(product,description):
|
|
109 |
partial_message = ""
|
110 |
audio_list = []
|
111 |
for chunk in output:
|
|
|
112 |
delta = chunk['choices'][0]['delta']
|
113 |
if 'content' in delta:
|
114 |
# audio_list.append([text_to_speech(delta.get('content', ''))])
|
|
|
59 |
repo_id="amir22010/fine_tuned_product_marketing_email_gemma_2_9b_q4_k_m", #custom fine tuned model
|
60 |
filename="unsloth.Q4_K_M.gguf", #model file name
|
61 |
cache_dir=os.path.abspath(os.getcwd()),
|
62 |
+
# n_ctx=2048,
|
63 |
+
# n_batch=126,
|
64 |
verbose=False
|
65 |
)
|
66 |
|
|
|
109 |
partial_message = ""
|
110 |
audio_list = []
|
111 |
for chunk in output:
|
112 |
+
print(chunk)
|
113 |
delta = chunk['choices'][0]['delta']
|
114 |
if 'content' in delta:
|
115 |
# audio_list.append([text_to_speech(delta.get('content', ''))])
|