Update app.py
Browse files
app.py
CHANGED
@@ -103,7 +103,7 @@ def main():
|
|
103 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
104 |
if st.button("Generate"):
|
105 |
tokenizer, model = load_models()
|
106 |
-
|
107 |
output = model.generate(input_ids, max_length=128)
|
108 |
st.success(tokenizer.decode(output[0], skip_special_tokens=True))
|
109 |
def change_photo_state():
|
|
|
103 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
104 |
if st.button("Generate"):
|
105 |
tokenizer, model = load_models()
|
106 |
+
st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
|
107 |
output = model.generate(input_ids, max_length=128)
|
108 |
st.success(tokenizer.decode(output[0], skip_special_tokens=True))
|
109 |
def change_photo_state():
|