Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def translate(prompt, top_p, temperature, repetition_penalty):
|
|
17 |
llm = VLLMOpenAI(
|
18 |
openai_api_key="EMPTY",
|
19 |
openai_api_base=os.getenv("VISTRAL_7B_MT_SERVER"),
|
20 |
-
model_name="nguyen1207/Vistral-7B-MT",
|
21 |
temperature=temperature,
|
22 |
top_p=top_p,
|
23 |
frequency_penalty=repetition_penalty,
|
|
|
17 |
llm = VLLMOpenAI(
|
18 |
openai_api_key="EMPTY",
|
19 |
openai_api_base=os.getenv("VISTRAL_7B_MT_SERVER"),
|
20 |
+
model_name="nguyen1207/Vistral-7B-MT-0.1",
|
21 |
temperature=temperature,
|
22 |
top_p=top_p,
|
23 |
frequency_penalty=repetition_penalty,
|