amir22010 commited on
Commit
df0ca26
·
verified ·
1 Parent(s): c9bd416

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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', ''))])