HariLogicgo commited on
Commit
0d9b89a
·
1 Parent(s): 0fd380e

import fixed

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -15,7 +15,8 @@ from fastapi.responses import RedirectResponse
15
  from pydantic import BaseModel
16
  from motor.motor_asyncio import AsyncIOMotorClient
17
  from bson.objectid import ObjectId
18
- from gridfs import AsyncIOMotorGridFSBucket
 
19
  from gradio import mount_gradio_app
20
  import uvicorn
21
  import logging
@@ -107,6 +108,7 @@ MONGODB_URL = os.getenv(
107
  client = AsyncIOMotorClient(MONGODB_URL)
108
  database = client.FaceSwap
109
  fs_bucket = AsyncIOMotorGridFSBucket(database)
 
110
  logger.info("MongoDB + GridFS initialized")
111
 
112
  # -------------------------------------------------
 
15
  from pydantic import BaseModel
16
  from motor.motor_asyncio import AsyncIOMotorClient
17
  from bson.objectid import ObjectId
18
+ from motor.motor_asyncio import AsyncIOMotorGridFSBucket
19
+
20
  from gradio import mount_gradio_app
21
  import uvicorn
22
  import logging
 
108
  client = AsyncIOMotorClient(MONGODB_URL)
109
  database = client.FaceSwap
110
  fs_bucket = AsyncIOMotorGridFSBucket(database)
111
+
112
  logger.info("MongoDB + GridFS initialized")
113
 
114
  # -------------------------------------------------
requirements.txt CHANGED
@@ -34,6 +34,7 @@ gradio
34
  huggingface_hub
35
  fastapi
36
  uvicorn
 
37
 
38
  # Optional but included in CodeFormer repo
39
  tb-nightly
 
34
  huggingface_hub
35
  fastapi
36
  uvicorn
37
+ motor
38
 
39
  # Optional but included in CodeFormer repo
40
  tb-nightly