Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import torch
 from PIL import Image
 from torchvision import transforms
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 import math
 from typing import Callable
 
@@ -434,7 +434,7 @@ class FluxParams:
     num_heads: int = 24
     depth: int = 19
     depth_single_blocks: int = 38
-    axes_dim: list = [16, 56, 56]
+    axes_dim: list[int] = field(default_factory=lambda: [16, 56, 56])
     theta: int = 10000
     qkv_bias: bool = True
     guidance_embed: bool = True
@@ -567,7 +567,7 @@ def get_schedule(
     return timesteps.tolist()
 
 def denoise(
-    model: Flux,
+    model: "Flux",
     img: Tensor,
     img_ids: Tensor,
     txt: Tensor,