radames commited on
Commit
4eeb78b
1 Parent(s): 8f77a4e

print reques

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -94,7 +94,6 @@ def generate(
94
  app = FastAPI()
95
  origins = [
96
  "http://huggingface.co",
97
- "localhost",
98
  ]
99
 
100
  app.add_middleware(
@@ -108,7 +107,7 @@ app.add_middleware(
108
 
109
  @app.middleware("http")
110
  async def validate_origin(request: Request, call_next):
111
- logging.info(f"Request origin: {request.headers.get('origin')}")
112
  if request.headers.get("origin") not in origins:
113
  raise HTTPException(status_code=403, detail="Forbidden")
114
  response = await call_next(request)
@@ -116,7 +115,7 @@ async def validate_origin(request: Request, call_next):
116
 
117
 
118
  @app.get("/image")
119
- async def generate_image(prompt: str, negative_prompt: str, seed: int = 2134213213):
120
  cached_img = database.check(prompt, negative_prompt, seed)
121
  if cached_img:
122
  logging.info(f"Image found in cache: {cached_img[0]}")
 
94
  app = FastAPI()
95
  origins = [
96
  "http://huggingface.co",
 
97
  ]
98
 
99
  app.add_middleware(
 
107
 
108
  @app.middleware("http")
109
  async def validate_origin(request: Request, call_next):
110
+ logging.info(f"Request origin: {request.headers}")
111
  if request.headers.get("origin") not in origins:
112
  raise HTTPException(status_code=403, detail="Forbidden")
113
  response = await call_next(request)
 
115
 
116
 
117
  @app.get("/image")
118
+ async def generate_image(prompt: str, negative_prompt: str = "", seed: int = 2134213213):
119
  cached_img = database.check(prompt, negative_prompt, seed)
120
  if cached_img:
121
  logging.info(f"Image found in cache: {cached_img[0]}")