Update app.py
Browse files
app.py
CHANGED
|
@@ -100,7 +100,7 @@ def main():
|
|
| 100 |
elif st.checkbox("Text Generation"):
|
| 101 |
st.subheader("Generate Text")
|
| 102 |
message = st.text_area("Enter the Text","Type please ..")
|
| 103 |
-
|
| 104 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
| 105 |
if st.button("Generate"):
|
| 106 |
st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
|
|
|
|
| 100 |
elif st.checkbox("Text Generation"):
|
| 101 |
st.subheader("Generate Text")
|
| 102 |
message = st.text_area("Enter the Text","Type please ..")
|
| 103 |
+
tokenizer, model = load_models()
|
| 104 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
| 105 |
if st.button("Generate"):
|
| 106 |
st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
|