Update app.py

app.py

@@ -4,7 +4,6 @@ from utils import load_model_pt, generate_image_stylegan, load_model_pkl, genera
 import random
 
 app = FastAPI()
-stylegan = load_model_pt("model_128.pt", model_type="stylegan")
 styleganv2 = load_model_pkl("styleganv2.pkl")
 gan = load_model_pt("generator_VanillaGAN.pt", model_type="vanillagan")
 
@@ -16,11 +15,6 @@ def root():
 def ping():
     return {"status": "pong"}
 
-@app.get("/generate/stylegan")
-def generate_stylegan():
-    image_stream = generate_image_stylegan(stylegan, steps=5, alpha=1.0)
-    return StreamingResponse(image_stream, media_type="image/png")
-
 @app.get("/generate/styleganv2")
 def generate_styleganv2(seed: int = Query(-1)):
     if seed == -1: 
@@ -32,4 +26,3 @@ def generate_styleganv2(seed: int = Query(-1)):
 def generate_vanillagan():
     image_stream = generate_image_vanillagan(gan)
     return StreamingResponse(image_stream, media_type="image/png")
-