Update app.py

app.py

@@ -32,13 +32,19 @@ if submit:
     do_sample=True, 
     max_length=50,
     top_p=0.99, 
-    num_return_sequences=2
+    num_return_sequences=1
   )
   
   st.subheader("Suggested sentences: ")
   
-  i = 0
-  for x in outputs:
-      out_text = ttokenizer.decode(x, skip_special_tokens=True)
-      i = i + 1
-      st.success(str(i) + '. ' + out_text)
+  out_text = ttokenizer.decode(outputs[0], skip_special_tokens=True)
+  st.success(str(i) + '. ' + out_text)
+  
+  c_text = " "
+  for x in input_text.split(" "):
+    if x in out_text.split(" "):
+      c_text = c_text + x
+    else:
+      c_text = c_text + """<p style="color:rgb(0,255,0);">""" + x + """</p>"""
+  
+  st.markdown(c_text)