Create model_b.py

model_wrappers/model_b.py

@@ -0,0 +1,8 @@
+# model_wrappers/model_b.py
+from transformers import pipeline
+
+model_b = pipeline("text-generation", model="distilgpt2")
+
+def run_model_b(prompt: str) -> str:
+    output = model_b(prompt, max_length=100, do_sample=False)
+    return output[0]["generated_text"]