Update app.py

app.py

@@ -23,8 +23,8 @@ def improve_image(img):
   resp_img = gr.processing_utils.decode_base64_to_image((resp_obj.json())['data'][0])
   return resp_img
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",use_auth_token=True)
-pipe = pipe.to("cuda")
+#pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",use_auth_token=True)
+#pipe = pipe.to("cuda")
 
 def generate(celebrity, setting):
   # ANSWER HERE
@@ -41,6 +41,6 @@ gr.Interface(
   outputs = gr.Image(type='pill'),
   allow_flagging="manual",
   flagging_options = ['Incorrect movie poster','Incorrect Actor','Other Problem'],
-  flagging_callback=hf_writer,
+  #flagging_callback=hf_writer,
   flagging_dir='/flagged_data'
 ).launch()