Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
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 |
|