Baptiste Canton commited on
Commit
c620a85
1 Parent(s): 9d49c48
Files changed (1) hide show
  1. captioner.py +6 -4
captioner.py CHANGED
@@ -54,8 +54,10 @@ async def startup_event():
54
  global app
55
  asyncio.create_task(asyncio.to_thread(load_model))
56
  # add gradio interface
57
- iface = gr.Interface(fn=captioner_gradapter, inputs="text", outputs=["text"], allow_flagging="never")
58
- app = gr.mount_gradio_app(app, iface, path="/gradio")
 
 
59
 
60
 
61
  async def captioner_gradapter(image_url):
@@ -66,9 +68,9 @@ async def captioner_gradapter(image_url):
66
  return caption
67
 
68
 
69
- @app.get("/")
70
  async def root():
71
- return {"message": "Hello World"}
72
 
73
 
74
  # the image url is passed in as a "url" tag in the json body
 
54
  global app
55
  asyncio.create_task(asyncio.to_thread(load_model))
56
  # add gradio interface
57
+ iface = gr.Interface(
58
+ fn=captioner_gradapter, inputs="text", outputs=["text"], allow_flagging="never"
59
+ )
60
+ app = gr.mount_gradio_app(app, iface, path="/")
61
 
62
 
63
  async def captioner_gradapter(image_url):
 
68
  return caption
69
 
70
 
71
+ """ @app.get("/")
72
  async def root():
73
+ return {"message": "Hello World"} """
74
 
75
 
76
  # the image url is passed in as a "url" tag in the json body