Update app.py

app.py

@@ -140,7 +140,7 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
         gr.Warning("Flagged potentially harmful output.")
         decoded = 'Flagged Output'
 
-    return decoded, gr.update(interactive=True)
+    return decoded #, gr.update(interactive=True)
 
 flagged_outputs = []
 
@@ -190,7 +190,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Generated Name", interactive=True),
-        gr.Button("Flag Output", interactive=False, elem_id="flag-button")
+        # gr.Button("Flag Output", interactive=False, elem_id="flag-button")
     ],
     live=False,
     title='Dungen Dev - Name Generator',
@@ -203,7 +203,9 @@ demo = gr.Interface(
         ["an evil character name", 30, 0.5]
     ],
     theme="default",
-    flagging_mode='never'
+    flagging_mode='manual',
+    flagging_options=["Incorrect", "Offensive/Hateful", "Harmful/Violent", "Illegal", "NSFW", "Other"],
+    flagging_dir='flagged_outputs'
 )
 
 demo.queue()