Update app.py

app.py

@@ -4,12 +4,13 @@ from peft import PeftConfig
 from transformers import AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 import gradio as gr
+from huggingface import spaces
 
 MODEL_PATH = "Ozaii/zephyr-bae"
 max_seq_length = 2048
 
-print("Attempting to load Zephyr... Cross your fingers! 🤞")
 @spaces.GPU
+print("Attempting to load Zephyr... Cross your fingers! 🤞")
 try:
     peft_config = PeftConfig.from_pretrained(MODEL_PATH)
     base_model_name = peft_config.base_model_name_or_path