Update inference

README.md

@@ -180,8 +180,7 @@ for text in generated:
 
 Output:
 ```text
-[TITLE]:
-  - 1: Macaroni and corn
+[TITLE]: Macaroni and corn
 [INGREDIENTS]:
   - 1: 2 c. macaroni
   - 2: 2 tbsp. butter
@@ -200,9 +199,8 @@ Output:
   - 6: Cook and stir until thickened and bubbly.
   - 7: Stir in corn and bacon.
   - 8: Pour over macaroni and mix well.
------------------------------------------------------------------------------------
-[TITLE]:
-  - 1: Grilled provolone and bacon sandwich
+----------------------------------------------------------------------------------------------------------------------------------
+[TITLE]: Grilled provolone and bacon sandwich
 [INGREDIENTS]:
   - 1: 2 slices provolone cheese
   - 2: 2 slices bacon
@@ -215,7 +213,7 @@ Output:
   - 4: Top with the other slice of bread.
   - 5: Heat a skillet over medium heat.
   - 6: Place the sandwich in the skillet and cook until the cheese is melted and the bread is golden brown.
------------------------------------------------------------------------------------
+----------------------------------------------------------------------------------------------------------------------------------
 ```
 
 ## Evaluation

src/prediction.py

@@ -96,8 +96,11 @@ 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)