Update app.py

app.py

@@ -38,7 +38,7 @@ def modify_image(image_url, prompt, mask_id=4):
       "redstonehero/ReV_Animated_Inpainting",
       torch_dtype=torch.float16)
 
-  pipeline.enable_model_cpu_offload()
+#  pipeline.enable_model_cpu_offload()
 
   image1 = pipeline(prompt=prompt,
                   num_inference_steps=24,
@@ -61,7 +61,7 @@ demo = gr.Interface(
     inputs=[
         gr.Image(type="filepath"), 
         gr.Textbox(label="Prompt"),
-        gr.Radio(["4", "5", "6"], label="Mask ID")
+        gr.Radio(["4", "6"], label="Mask ID")
     ],
     outputs=gr.Image()
 )