Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def clean_generated_text(text):
|
|
32 |
clean_text = re.sub(r'</s>$', '', text).strip()
|
33 |
return clean_text
|
34 |
|
35 |
-
def generate_text(prompt, temperature=0.9, max_new_tokens=
|
36 |
temperature = max(temperature, 1e-2)
|
37 |
generate_kwargs = dict(
|
38 |
temperature=temperature,
|
|
|
32 |
clean_text = re.sub(r'</s>$', '', text).strip()
|
33 |
return clean_text
|
34 |
|
35 |
+
def generate_text(prompt, temperature=0.9, max_new_tokens=1512, top_p=0.95, repetition_penalty=1.0):
|
36 |
temperature = max(temperature, 1e-2)
|
37 |
generate_kwargs = dict(
|
38 |
temperature=temperature,
|