Spaces:
Sleeping
Sleeping
update
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ from fastapi import FastAPI, Query
|
|
| 2 |
from fastapi.middleware.cors import CORSMiddleware
|
| 3 |
from fastapi.responses import StreamingResponse
|
| 4 |
from utils import load_model_pkl, generate_image_from_pkl, generate_image_from_onnx
|
|
|
|
| 5 |
|
| 6 |
app = FastAPI()
|
| 7 |
|
|
@@ -28,7 +29,9 @@ def generate_stylegan_onnx():
|
|
| 28 |
return StreamingResponse(image_stream, media_type="image/png")
|
| 29 |
|
| 30 |
@app.get("/generate/styleganv2")
|
| 31 |
-
def generate_styleganv2(seed: int = Query(
|
|
|
|
|
|
|
| 32 |
image_stream = generate_image_from_pkl(styleganv2, seed=seed, trunc=1)
|
| 33 |
return StreamingResponse(image_stream, media_type="image/png")
|
| 34 |
|
|
|
|
| 2 |
from fastapi.middleware.cors import CORSMiddleware
|
| 3 |
from fastapi.responses import StreamingResponse
|
| 4 |
from utils import load_model_pkl, generate_image_from_pkl, generate_image_from_onnx
|
| 5 |
+
from random import randint
|
| 6 |
|
| 7 |
app = FastAPI()
|
| 8 |
|
|
|
|
| 29 |
return StreamingResponse(image_stream, media_type="image/png")
|
| 30 |
|
| 31 |
@app.get("/generate/styleganv2")
|
| 32 |
+
def generate_styleganv2(seed: int = Query(None)):
|
| 33 |
+
if seed is None:
|
| 34 |
+
seed = randint(0, 2**32 - 1)
|
| 35 |
image_stream = generate_image_from_pkl(styleganv2, seed=seed, trunc=1)
|
| 36 |
return StreamingResponse(image_stream, media_type="image/png")
|
| 37 |
|