Added longer examples.

app.py

@@ -5,10 +5,23 @@ from transformers import pipeline
 from transformers.pipelines import PIPELINE_REGISTRY, FillMaskPipeline
 from transformers import AutoModelForMaskedLM
 
-# unmasker = pipeline("temp-scale", model="anferico/bert-for-patents")
-examples = [['A crustless [MASK] made from two slices of baked bread.', 1.2],
-           ['The invention provides a method for altering or modifying [MASK] of one or more gene products.', 1.1],
-           ['The graphite [MASK] is composed of a two-dimensional hexagonal lattice of carbon atoms.', 1.4]]
+ex_str1 = "A crustless sandwich made from two slices of baked bread. The sandwich includes first and second matching " \
+          "crustiess bread pieces. The bread pieces have the same general outer shape defined by an outer periphery " \
+          "with central portions surrounded by an outer peripheral area, the bread pieces being at least partially " \
+          "crimped together at the outer peripheral area."
+
+ex_str2 = "The present disclosure provides a DNA-targeting RNA that comprises a targeting sequence and, together with" \
+          " a modifying polypeptide, provides for site-specific modification of a target DNA and/or a polypeptide" \
+          " associated with the target DNA. "
+
+ex_str3 = "The graphite plane is composed of a two-dimensional hexagonal lattice of carbon atoms and the plate has a " \
+          "length and a width parallel to the graphite plane and a thickness orthogonal to the graphite plane with at " \
+          "least one of the length, width, and thickness values being 100 nanometers or smaller. "
+
+examples = [[ex_str1, 1.2, 50],
+           [ex_str2, 1.5, 10],
+           [ex_str3, 1.4, 20]]
+
 
 def add_mask(text, size=1):
     split_text = text.split()
@@ -138,17 +151,20 @@ def unmask(text, temp, rounds):
         new_token = score_to_str[score]
         split_text[mask_pos] = new_token
         text = ' '.join(split_text)
-    return text
+    text = list(text)
+    text[0] = text[0].upper()
+    return ''.join(text)
+
 
-textbox = gr.Textbox(label="Type language here", lines=5)
-textbox2 = gr.Textbox(placeholder="", lines=4)
+textbox = gr.Textbox(label="Example prompts", lines=5)
+output_textbox = gr.Textbox(placeholder="", lines=4)
 temp_slider = gr.Slider(1.0, 2.0, value=1.0, label='Creativity')
-edit_slider = gr.Slider(1, 50, step=1, value=1.0, label='Number of edits')
+edit_slider = gr.Slider(1, 150, step=5, value=1.0, label='Number of edits')
 
 demo = gr.Interface(
     fn=unmask,
     inputs=[textbox, temp_slider, edit_slider],
-    outputs=[textbox2],
+    outputs=[output_textbox],
     examples=examples,
 )