Spaces:
Running
on
T4
Running
on
T4
Commit
·
0d9b89a
1
Parent(s):
0fd380e
import fixed
Browse files- app.py +3 -1
- 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
|
|
|
|
| 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
|