woods-today commited on
Commit
7eb427a
·
1 Parent(s): 3767788

Working on it

Browse files
Files changed (2) hide show
  1. requirements-fastapi.txt +5 -2
  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 get_statistics_training():
14
- content = ["HI"]
15
- return content
 
 
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]