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