Update app.py

app.py

@@ -78,6 +78,9 @@ def train_model():
 def generate_text(prompt):
     global tokenizer  # Ensure we use the global tokenizer variable
     
+    if tokenizer is None:
+        return "Tokenizer not initialized. Please train the model first."
+    
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     
     # Adjust generation parameters for better quality output