Update app.py
Browse files
app.py
CHANGED
|
@@ -9,12 +9,15 @@ def generate_text(prompt):
|
|
| 9 |
return generator(
|
| 10 |
prompt,
|
| 11 |
max_length=200,
|
| 12 |
-
temperature=0.
|
| 13 |
-
top_k=50,
|
| 14 |
-
top_p=0.9
|
|
|
|
|
|
|
| 15 |
)[0]["generated_text"]
|
| 16 |
|
| 17 |
|
|
|
|
| 18 |
interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
|
| 19 |
interface.launch()
|
| 20 |
|
|
|
|
| 9 |
return generator(
|
| 10 |
prompt,
|
| 11 |
max_length=200,
|
| 12 |
+
temperature=0.7, # Lower temperature for more coherent responses
|
| 13 |
+
top_k=50, # Limits sampling pool to the top 50 tokens
|
| 14 |
+
top_p=0.9, # Limits sampling to 90% cumulative probability
|
| 15 |
+
repetition_penalty=1.2, # Penalizes repeated phrases (increase if repetition continues)
|
| 16 |
+
do_sample=True # Enables sampling for added randomness
|
| 17 |
)[0]["generated_text"]
|
| 18 |
|
| 19 |
|
| 20 |
+
|
| 21 |
interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
|
| 22 |
interface.launch()
|
| 23 |
|