Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,5 +16,5 @@ prompt = st.text_input("Enter your message:")
|
|
16 |
if st.button("Run"):
|
17 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
18 |
outputs = model.generate(inputs)
|
19 |
-
response = tokenizer.decode(outputs, skip_special_tokens=False)
|
20 |
st.text_area("Long Text Box", response, height=300)
|
|
|
16 |
if st.button("Run"):
|
17 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
18 |
outputs = model.generate(inputs)
|
19 |
+
response = tokenizer.decode(outputs[0], skip_special_tokens=False)
|
20 |
st.text_area("Long Text Box", response, height=300)
|