[ / ] Test: Add upscale to generation

app.py

@@ -144,7 +144,7 @@ def generate(
     upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
 
     try:
-
+        gr.Info("Generating image...")
         latents = pipe(
             prompt = prompt,
             negative_prompt = negative_prompt,
@@ -159,6 +159,7 @@ def generate(
 
         upscaled_latents = upscale(latents, "nearest-exact", 2.0)
 
+        gr.Info("Upscaling...")
         img = upscaler_pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
@@ -180,7 +181,7 @@ def generate(
 
 if torch.cuda.is_available():
     pipe = load_pipeline(MODEL)
-    print("Loaded on Device!")
+    gr.Info("Loaded on Device!")
 else:
     pipe = None