M1ck1 commited on
Commit
2a0790c
·
verified ·
1 Parent(s): fbcf2f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -9,12 +9,15 @@ def generate_text(prompt):
9
  return generator(
10
  prompt,
11
  max_length=200,
12
- temperature=0.8, # Adjust between 0.7 - 1.0
13
- top_k=50, # Try values between 30 - 100
14
- top_p=0.9 # Adjust between 0.8 - 1.0
 
 
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