Spaces:
Build error
Build error
woods-today
commited on
Commit
·
7eb427a
1
Parent(s):
3767788
Working on it
Browse files- requirements-fastapi.txt +5 -2
- routers/training.py +4 -3
requirements-fastapi.txt
CHANGED
@@ -8,5 +8,8 @@ donut-python
|
|
8 |
fastapi==0.95.2
|
9 |
uvicorn[standard]
|
10 |
python-multipart
|
11 |
-
diffusers
|
12 |
-
torch
|
|
|
|
|
|
|
|
8 |
fastapi==0.95.2
|
9 |
uvicorn[standard]
|
10 |
python-multipart
|
11 |
+
diffusers==0.10.2
|
12 |
+
torch
|
13 |
+
scipy
|
14 |
+
ftfy
|
15 |
+
accelerate
|
routers/training.py
CHANGED
@@ -10,6 +10,7 @@ import utils
|
|
10 |
router = APIRouter()
|
11 |
|
12 |
@router.get("/hi")
|
13 |
-
async def
|
14 |
-
|
15 |
-
|
|
|
|
10 |
router = APIRouter()
|
11 |
|
12 |
@router.get("/hi")
|
13 |
+
async def hifunction():
|
14 |
+
prompt = "a photograph of an astronaut riding a horse"
|
15 |
+
image = pipe(prompt).images[0]
|
16 |
+
return [image]
|