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)
|