Spaces:
Runtime error
Runtime error
reshinthadithyan
commited on
Commit
·
8cd2bac
1
Parent(s):
a83a01c
app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ def generate_text(prompt):
|
|
| 24 |
)
|
| 25 |
|
| 26 |
outputs = tokenizer.batch_decode(output_seq, skip_special_tokens=False)
|
|
|
|
| 27 |
return outputs
|
| 28 |
st.set_page_config(
|
| 29 |
page_title= "Code Representation Learning",
|
|
|
|
| 24 |
)
|
| 25 |
|
| 26 |
outputs = tokenizer.batch_decode(output_seq, skip_special_tokens=False)
|
| 27 |
+
outputs = outputs[0].split("<bash>")[-1]
|
| 28 |
return outputs
|
| 29 |
st.set_page_config(
|
| 30 |
page_title= "Code Representation Learning",
|