Spaces:

davidizzle
/

LIA_LLM_PoC

Running

davidizzle commited on 20 days ago

Commit

2190f14

1 Parent(s): 79357a0

Sorry

Files changed (1) hide show

app.py CHANGED Viewed

@@ -128,5 +128,5 @@ if st.button("Generate"):
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     st.markdown("### ✨ Output:")
-    final_thought = result.split("<think>\n")[1:]
     st.write(final_thought)

     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     st.markdown("### ✨ Output:")
+    final_thought = result.split("</think>\n")[1:]
     st.write(final_thought)