arieridwans commited on
Commit
13184eb
1 Parent(s): 0a2a6c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ if st.button("Generate Output"):
30
  return_attention_mask=False,
31
  padding=True,
32
  truncation=True)
33
- result = inference_model.generate(**input, repetition_penalty=1.2, max_length=1024)
34
  output = tokenizer.batch_decode(result, skip_special_tokens=True)[0]
35
  st.text("Generated Result:")
36
  st.write(output)
 
30
  return_attention_mask=False,
31
  padding=True,
32
  truncation=True)
33
+ result = model.generate(**input, repetition_penalty=1.2, max_length=1024)
34
  output = tokenizer.batch_decode(result, skip_special_tokens=True)[0]
35
  st.text("Generated Result:")
36
  st.write(output)