Update model_wrappers/model_b.py

model_wrappers/model_b.py

@@ -11,4 +11,5 @@ def run_model_b(prompt: str) -> str:
                      temperature=0.8,
                      top_k=50,
                      top_p=0.95)
-    return output[0]["generated_text"]
+    return output[0]["generated_text"]
+