Update app.py

app.py

@@ -9,6 +9,7 @@ import os
 
 # required for stable difussion
 auth_token = os.environ.get("auth_token") #in secret space
+hf_writer = gr.HuggingFaceDatasetSaver(auth_token, "flagged-movie-poster-celebrity-swap")
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -63,11 +64,15 @@ def generate(celebrity, movie, guidance, improve_flag, scale): # add scale as va
     image = improve_image(image, scale=scale)
   return image
 
-movie_options = ["James Bond", "Snatch", "Saving Private Ryan", "Miss Congeniality", "Scarface", "Avatar", "Top Gun"]
-title = "Movie Poster Celebrity Swap"
-description = "Write the name of a celebrity, and pick a movie from the dropdown menu.\
-               It will generate a new movie poster (inspired by your chosen movie)\
-               with the chosen celebrity in it. See below for explanation of the\
+movie_options = movie_options = ["Casino Royale (2006)",
+                "Quantum of Solace (2008)",
+                "Skyfall (2012)",
+                "Spectre (2015)",
+                "No Time To Die (2021)"]
+title = "Find the new James Bond"
+description = "Write the name of a celebrity, and pick a Bond movie from the dropdown menu.\
+               This will generate a new movie poster (inspired by the chosen movie)\
+               with the new celebrity in it. See below for explanation of the\
                input variables."
 article= "Inputs explained: \n Guidance: the lower, the more random the output\
           image. Improve and scale: if selected, the image will be refined\
@@ -75,8 +80,8 @@ article= "Inputs explained: \n Guidance: the lower, the more random the output\
 
 demo = gr.Interface(
     fn=generate,
-    inputs=[gr.Textbox(value="Daniel Craig"), 
-            gr.Dropdown(movie_options, value="Saving Private Ryan"), 
+    inputs=[gr.Textbox(value="Emma Watson"), 
+            gr.Dropdown(movie_options, value="Skyfall (2012)"), 
             gr.Slider(1, 20, value=7.5, step=0.5),
             gr.Checkbox(label="Improve and scale? (extra time)"),
             gr.Slider(1, 3, value=1, step=0.5)
@@ -84,7 +89,10 @@ demo = gr.Interface(
     outputs='image',
     title=title,
     description=description,
-    article=article
+    article=article,
+    allow_flagging="manual",
+    flagging_options=["wrong celebrity", "wrong movie", "face not shown", "nonsensical output"],
+    flagging_callback=hf_writer
 ) 
 
 demo.launch()