Spaces:
Sleeping
Sleeping
arieridwans
commited on
Commit
•
13184eb
1
Parent(s):
0a2a6c5
Update app.py
Browse files
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 =
|
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)
|