Fixing error

app.py

@@ -3,7 +3,6 @@ import tiktoken
 import gradio as gr
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
-from model import GPTModel
 import spaces
 
 n_layers = 24
@@ -16,6 +15,7 @@ dropout = 0.0
 
 @spaces.GPU
 def load_model():
+    from model import GPTModel
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     tokenizer = tiktoken.encoding_for_model('gpt2')
     model_path = hf_hub_download('cbspace/gpt', 'model.safetensors')