0x7o commited on
Commit
5415a35
Β·
verified Β·
1 Parent(s): 7f2465d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ if torch.cuda.is_available():
10
 
11
  @spaces.GPU
12
  def predict(prompt, temperature, max_length):
13
- return pipe(prompt, temperature=temperature, max_length=max_length)[0]["generated_text"]
14
 
15
  demo = gr.Interface(
16
  fn=predict,
 
10
 
11
  @spaces.GPU
12
  def predict(prompt, temperature, max_length):
13
+ return pipe(prompt, temperature=temperature, max_length=max_length, top_p=0.95, top_k=50, do_sample=True)[0]["generated_text"]
14
 
15
  demo = gr.Interface(
16
  fn=predict,