Update inference

README.md

@@ -171,9 +171,12 @@ for text in generated:
             section = section.replace("directions:", "")
             headline = "DIRECTIONS"
         
-        section_info = [f"  - {i+1}: {info.strip().capitalize()}" for i, info in enumerate(section.split("--"))]
-        print(f"[{headline}]:")
-        print("\n".join(section_info))
+        if headline == "TITLE":
+            print(f"[{headline}]: {section.strip().capitalize()}")
+        else:
+            section_info = [f"  - {i+1}: {info.strip().capitalize()}" for i, info in enumerate(section.split("--"))]
+            print(f"[{headline}]:")
+            print("\n".join(section_info))
 
     print("-" * 130)
 ```