BigSalmon commited on
Commit
aa2f0b6
·
1 Parent(s): a5cd868

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -28,7 +28,7 @@ def process(text: str,
28
  "top_k": top_k,
29
  "top_p": top_p,
30
  "temperature": temp,
31
- "num_return_sequences": 10,
32
  "repetition_penalty": 2.0,
33
  },
34
  "options": {
@@ -97,7 +97,6 @@ if st.button("Run"):
97
  for error in result["error"]:
98
  st.write(f'{error}')
99
  else:
100
- result = result[0]["generated_text"]
101
- st.write(result.replace("\n", " \n"))
102
- st.text("Thai 🇹🇭 to English 🇬🇧 translation")
103
- st.write(result)
 
28
  "top_k": top_k,
29
  "top_p": top_p,
30
  "temperature": temp,
31
+ "num_return_sequences": num_return_sequences,
32
  "repetition_penalty": 2.0,
33
  },
34
  "options": {
 
97
  for error in result["error"]:
98
  st.write(f'{error}')
99
  else:
100
+ for i in range(num_return_sequences):
101
+ result = result[i]["generated_text"]
102
+ st.write(result.replace("\n", " \n"))