Update app.py

app.py

@@ -58,7 +58,6 @@ css = """
 
 url = "http://localhost:7860"
 
-@spaces.GPU
 def generate(merged_image, total_mask, original_image, add_color_image, add_edge_mask, remove_edge_mask, fill_mask, add_prop_image, positive_prompt, negative_prompt, fine_edge, fix_perspective, grow_size, edge_strength, color_strength, local_strength, seed, steps, cfg):
     print("prompt is:", positive_prompt)
     print("other parameters:", negative_prompt, fine_edge, fix_perspective, grow_size, edge_strength, color_strength, local_strength, seed, steps, cfg)
@@ -127,6 +126,7 @@ def generate(merged_image, total_mask, original_image, add_color_image, add_edge
     res_base64 = tensor_to_base64(final_image)
     return res_base64
 
+@spaces.GPU
 def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_size, edge_strength, color_strength, local_strength, seed, steps, cfg):
     merged_image = x['from_frontend']['img']
     total_mask = x['from_frontend']['total_mask']
@@ -267,31 +267,31 @@ def get_root_url(
 import gradio.route_utils 
 gr.route_utils.get_root_url = get_root_url
 
-@app.post("/magic_quill/generate_image")
-async def generate_image(request: Request):
-    data = await request.json()
-    res = generate(
-        data["merged_image"],
-        data["total_mask"],
-        data["original_image"],
-        data["add_color_image"],
-        data["add_edge_mask"],
-        data["remove_edge_mask"],
-        data["fill_mask"],
-        data["add_prop_image"],
-        data["positive_prompt"],
-        data["negative_prompt"],
-        data["fine_edge"],
-        data["fix_perspective"],
-        data["grow_size"],
-        data["edge_strength"],
-        data["color_strength"],
-        data["local_strength"],
-        data["seed"],
-        data["steps"],
-        data["cfg"]
-    )
-    return {'res': res}
+# @app.post("/magic_quill/generate_image")
+# async def generate_image(request: Request):
+#     data = await request.json()
+#     res = generate(
+#         data["merged_image"],
+#         data["total_mask"],
+#         data["original_image"],
+#         data["add_color_image"],
+#         data["add_edge_mask"],
+#         data["remove_edge_mask"],
+#         data["fill_mask"],
+#         data["add_prop_image"],
+#         data["positive_prompt"],
+#         data["negative_prompt"],
+#         data["fine_edge"],
+#         data["fix_perspective"],
+#         data["grow_size"],
+#         data["edge_strength"],
+#         data["color_strength"],
+#         data["local_strength"],
+#         data["seed"],
+#         data["steps"],
+#         data["cfg"]
+#     )
+#     return {'res': res}
 
 @app.post("/magic_quill/process_background_img")
 async def process_background_img(request: Request):