Spaces:

mosama
/

Qwen2.5-0.5B-Base-Model

Sleeping

mosama commited on Jan 4

Commit

3effc74

verified ·

1 Parent(s): 572fdfd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ if prompt:
             top_k=20
         )
         print("Generated Response!")
-        response = tokenizer.decode(generated_ids, skip_special_tokens=True)[0]
     # Display assistant response in chat message container
     with st.chat_message("assistant"):

             top_k=20
         )
         print("Generated Response!")
+        response = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
     # Display assistant response in chat message container
     with st.chat_message("assistant"):