Update generator.py

generator.py

@@ -0,0 +1,16 @@
+## File: `generator.py`
+
+```python
+"""Optional separate generator module if you want to swap models easily."""
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+
+class Generator:
+    def __init__(self, model_name: str = "google/flan-t5-small"):
+        self.model_name = model_name
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
+        self.pipeline = pipeline("text2text-generation", model=self.model, tokenizer=self.tokenizer)
+
+    def generate(self, prompt: str, max_length: int = 256):
+        out = self.pipeline(prompt, max_length=max_length, do_sample=False)
+        return out[0]['generated_text'] if out else ""