Update app.py

app.py

@@ -34,7 +34,7 @@ class FluxGenerator:
             offload=self.offload,
         )
         self.toonmage_model = ToonMagePipeline(self.model, 'cuda', weight_dtype=torch.bfloat16)
-        self.toonmahe_model.load_pretrain()
+        self.toonmage_model.load_pretrain()
 
 
 flux_generator = FluxGenerator()