Spaces:

davidizzle
/

LIA_LLM_PoC

Running

davidizzle commited on 20 days ago

Commit

466be59

1 Parent(s): ceb2dbe

Omit reasoning

Files changed (1) hide show

app.py CHANGED Viewed

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

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