add uvicorn server

webaverse · Oct 26, 2022 · 07f14ac · 07f14ac
1 parent 3becbfe
commit 07f14ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -6,6 +6,7 @@
 
 from models.blip import blip_decoder
 from fastapi import FastAPI
+import uvicorn
 
 
 device = 'cuda:0'
@@ -52,3 +53,7 @@ async def exec_image_captioning(img_url: str):
 			return {'caption': caption[0]}
 	except Exception as e:
 		return {'Error': e}
+
+
+if __name__ == '__main__':
+	uvicorn.run(app, host='0.0.0.0', port='8080')
diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,4 @@ transformers==4.15.0
 fairscale==0.4.4
 pycocoevalcap
 fastapi
+uvicorn
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,3 +3,4 @@ transformers==4.15.0 @@
     fairscale==0.4.4
     pycocoevalcap
     fastapi
+    uvicorn