Update app.py

app.py

@@ -9,6 +9,11 @@ def generate(celebrity, movie,setting):
   image = pipe(prompt).images[0]  
   return image
 
+import os
+
+HF_TOKEN = os.getenv('sai')
+hf_writer =  gr.HuggingFaceDatasetSaver(HF_TOKEN, "movie-poster")
+
 gr.Interface(
   fn = generate,  
     inputs=[
@@ -17,5 +22,8 @@ gr.Interface(
         gr.inputs.Textbox(label="Setting")               
     ],
     outputs= gr.outputs.Image(label="Generated movie poster",type='pil'),
-    title="Movie Poster Generator"  
+    title="Movie Poster Generator"  ,
+    allow_flagging="manual",
+    flagging_options=["Good", "Better", "Worst"],
+    flagging_callback=hf_writer
 ).launch(share=True)