Pclanglais commited on
Commit
5c73fde
·
verified ·
1 Parent(s): 3343601

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -28,7 +28,7 @@ repetition_penalty=1.7
28
 
29
  model_name = "Inagua/code-model-2"
30
 
31
- llm = LLM(model_name, max_model_len=4096)
32
 
33
 
34
 
@@ -139,6 +139,7 @@ class MistralChatBot:
139
  prompts = [detailed_prompt]
140
  outputs = llm.generate(prompts, sampling_params, use_tqdm = False)
141
  generated_text = outputs[0].outputs[0].text
 
142
  fiches_html = ""
143
  return generated_text, fiches_html
144
 
 
28
 
29
  model_name = "Inagua/code-model-2"
30
 
31
+ llm = LLM(model_name, max_model_len=8192)
32
 
33
 
34
 
 
139
  prompts = [detailed_prompt]
140
  outputs = llm.generate(prompts, sampling_params, use_tqdm = False)
141
  generated_text = outputs[0].outputs[0].text
142
+ print(generated_text)
143
  fiches_html = ""
144
  return generated_text, fiches_html
145