fix error

stylegan_generator.py

@@ -1,16 +1,15 @@
 import os
-import io
 import math
 import numpy as np
 import onnxruntime as ort
 from PIL import Image
-import gradio as gr
 
 LATENT_FEATURES = 512
 RESOLUTION = 256
 LAST_INDEX = math.log2(RESOLUTION) - 2
 
 MODEL_PATH = os.path.join("model", "batik_stylegan.onnx")
+model = ort.InferenceSession(MODEL_PATH)
 
 alpha = np.array([1.0], dtype=np.float32)
 steps = np.array([LAST_INDEX], dtype=np.int64)
@@ -30,4 +29,4 @@ def generate_stylegan():
     image = np.transpose(image, (1, 2, 0))
     pil_img = Image.fromarray(image, 'RGB')
     
-    return pil_img.resize((512, 512), Image.LANCZOS)
+    return pil_img.resize((512, 512), Image.LANCZOS)