transformers-CFG-JSON-demo

Sleeping

joaogante commited on Feb 7, 2023

Commit

0243d8b

1 Parent(s): ade3ffb

button text

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ if __name__ == "__main__":
         with gr.Row():
             with gr.Column():
                 prompt = gr.Textbox(label="Prompt", lines=3, value="Today is")
-                button = gr.Button(f"Generate with {MODEL_NAME}")
             with gr.Column():
                 highlighted_text = gr.HighlightedText(
                     label="Highlighted generation",

         with gr.Row():
             with gr.Column():
                 prompt = gr.Textbox(label="Prompt", lines=3, value="Today is")
+                button = gr.Button(f"Generate with {MODEL_NAME}, using sampling!")
             with gr.Column():
                 highlighted_text = gr.HighlightedText(
                     label="Highlighted generation",