dipankardas011 commited on
Commit
aa4a0d4
·
unverified ·
1 Parent(s): ff29d15

Added url entry mode

Browse files

Signed-off-by: Dipankar Das <dipankardas0115@gmail.com>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ async def docs_redirect():
16
  return RedirectResponse(url='/docs')
17
 
18
  @app.get("/generate")
19
- def generate():
20
- url = "https://www.ilankelman.org/stopsigns/australia.jpg"
21
  image = Image.open(requests.get(url, stream=True).raw)
22
 
23
  inputs = processor(images=image, return_tensors="pt")
 
16
  return RedirectResponse(url='/docs')
17
 
18
  @app.get("/generate")
19
+ def generate(url: str):
20
+ # url = "https://www.ilankelman.org/stopsigns/australia.jpg"
21
  image = Image.open(requests.get(url, stream=True).raw)
22
 
23
  inputs = processor(images=image, return_tensors="pt")