Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ def generate_text(prompt, max_length, temperature):
|
|
| 14 |
max_length=max_length,
|
| 15 |
temperature=temperature,
|
| 16 |
do_sample=True,
|
| 17 |
-
repetition_penalty=1.
|
| 18 |
)
|
| 19 |
return generated[0]['generated_text']
|
| 20 |
except Exception as e:
|
|
|
|
| 14 |
max_length=max_length,
|
| 15 |
temperature=temperature,
|
| 16 |
do_sample=True,
|
| 17 |
+
repetition_penalty=1.4
|
| 18 |
)
|
| 19 |
return generated[0]['generated_text']
|
| 20 |
except Exception as e:
|