LiuZichen commited on
Commit
cfb64d9
·
verified ·
1 Parent(s): 9a41f03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -39
app.py CHANGED
@@ -230,45 +230,6 @@ def get_root_url(request: Request, route_path: str, root_path: str | None):
230
  return root_path
231
  gr.route_utils.get_root_url = get_root_url
232
 
233
- @app.post("/magic_quill/generate_image")
234
- async def generate_image(request: Request):
235
- data = await request.json()
236
-
237
- # 1. Extract headers from the FastAPI request object
238
- forward_headers = get_zerogpu_headers(request.headers)
239
-
240
- try:
241
- # 2. Instantiate client with headers
242
- client = Client(BACKEND_URL, headers=forward_headers)
243
-
244
- res = await run_in_threadpool(
245
- client.predict,
246
- data["merged_image"],
247
- data["total_mask"],
248
- data["original_image"],
249
- data["add_color_image"],
250
- data["add_edge_mask"],
251
- data["remove_edge_mask"],
252
- data["fill_mask"],
253
- data["add_prop_image"],
254
- data["positive_prompt"],
255
- data["negative_prompt"],
256
- data["fine_edge"],
257
- data["fix_perspective"],
258
- data["grow_size"],
259
- data["edge_strength"],
260
- data["color_strength"],
261
- data["local_strength"],
262
- data["seed"],
263
- data["steps"],
264
- data["cfg"],
265
- api_name="/generate"
266
- )
267
- return {'res': res}
268
- except Exception as e:
269
- print(f"Error in backend generation: {e}")
270
- return {'error': str(e)}
271
-
272
  @app.post("/magic_quill/process_background_img")
273
  async def process_background_img(request: Request):
274
  img = await request.json()
 
230
  return root_path
231
  gr.route_utils.get_root_url = get_root_url
232
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
233
  @app.post("/magic_quill/process_background_img")
234
  async def process_background_img(request: Request):
235
  img = await request.json()