raconteur25 commited on
Commit
69e06b4
·
verified ·
1 Parent(s): f3bed9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -27,11 +27,10 @@ Speak with the voice of a storm that does not beg the sky for mercy.
27
 
28
  generator = pipeline(
29
  "text-generation",
30
- model="tiiuae/falcon-7b-instruct",
31
- use_auth_token=True,
32
  max_length=300,
33
  temperature=0.5,
34
- top_p=0.85, # slightly restrict random topics
35
  repetition_penalty=1.2
36
  )
37
 
 
27
 
28
  generator = pipeline(
29
  "text-generation",
30
+ model="tiiuae/falcon-rw-1b",
 
31
  max_length=300,
32
  temperature=0.5,
33
+ top_p=0.85,
34
  repetition_penalty=1.2
35
  )
36