Spaces:

davidizzle
/

LIA_LLM_PoC

Sleeping

davidizzle commited on Apr 19

Commit

a90ecdc

1 Parent(s): 1eb2d22

Riky

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,5 +131,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(result)