Update app.py

app.py

@@ -29,9 +29,13 @@ torch.set_float32_matmul_precision("highest")
 
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
 
-text_to_image = keras_hub.models.StableDiffusion3TextToImage.from_preset(
-    "stable_diffusion_3_medium", width=768, height=768, dtype="bfloat16"
-)
+def load_model():
+    text_to_image = keras_hub.models.StableDiffusion3TextToImage.from_preset(
+        "stable_diffusion_3_medium", width=768, height=768, dtype="bfloat16"
+    )
+    return text_to_image
+
+text_to_image = load_model()
 
 code = r'''
 import paramiko