spaces error fix

app.py

@@ -54,6 +54,7 @@ def fused_sim(a:Image.Image,b:Image.Image,α=.5):
 @spaces.GPU
 def load_models():
     from unsloth import FastLanguageModel
+    global base, tok, lora
     bnb_cfg = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_use_double_quant=True)
     print("Loading BASE …")
     base, tok = FastLanguageModel.from_pretrained(
@@ -65,7 +66,7 @@ def load_models():
     lora, _ = FastLanguageModel.from_pretrained(
         ADAPTER_DIR, max_seq_length=2048,
         load_in_4bit=True, quantization_config=bnb_cfg, device_map="auto")
-    return base, tok, lora
+    print("✔ models loaded")
 
 
 base, tok, lora = load_models()
@@ -93,7 +94,8 @@ def compare(desc):
     img_base, svg_base = draw(base, desc)
     img_lora, svg_lora = draw(lora, desc)
     # sim = (fused_sim(img_lora, img_base) if img_base and img_lora else float("nan"))
-    caption = "Thanks for trying our model 😊"
+
+    caption = "Thanks for trying our model 😊\nIf you don't see an image for the base or GRPO model that means it didn't generate a valid SVG!"
     return img_base, img_lora, caption, svg_base, svg_lora
 
 with gr.Blocks(css="body{background:#111;color:#eee}") as demo: