Update app.py

app.py

@@ -29,22 +29,23 @@ if submit:
     do_sample=True, 
     max_length=50,
     top_p=0.99, 
-    top_k=70,
-    num_return_sequences=1
+    top_k=50,
+    num_return_sequences=3
   )
   
-  st.subheader("Suggested sentence: ")
-  
-  out_text = ttokenizer.decode(outputs[0], skip_special_tokens=True)
-  st.success(out_text.capitalize())
-  
-  st.markdown("### Edited sentence:")
-  c_text = ""
-  for x in out_text.lower().split(" "):
-    if x in input_text.lower().split(" "):
-      c_text = c_text + x + " "
-    else:
-      c_text = c_text + '<span style="font-weight:bold; color:rgb(150,255,100);">' + x + '</span>' + " "
-      
-  ct = c_text.capitalize()
-  st.markdown(str(ct), unsafe_allow_html=True)
+  for x in outputs:
+    st.subheader("Most probable: ")
+    
+    out_text = ttokenizer.decode(x, skip_special_tokens=True)
+    st.success(out_text.capitalize())
+    
+    st.markdown("### Edited sentences:")
+    c_text = ""
+    for x in out_text.lower().split(" "):
+      if x in input_text.lower().split(" "):
+        c_text = c_text + x + " "
+      else:
+        c_text = c_text + '<span style="font-weight:bold; color:rgb(150,255,100);">' + x + '</span>' + " "
+        
+    ct = c_text.capitalize()
+    st.markdown(str(ct), unsafe_allow_html=True)